diff --git a/tests/test_theme.py b/tests/test_theme.py
index 23f9b5a..31de80f 100644
--- a/tests/test_theme.py
+++ b/tests/test_theme.py
@@ -19,8 +19,8 @@ def test_theme_view(mocked_random, app, admin_user, fake_themes):
     assert Variable.objects.filter(name='theme')[0].value == 'alfortville'
     assert Variable.objects.filter(name='foo')[0].value == 'bar'
     assert resp.location == '/theme/'
-    assert "The theme has been changed" in dict(resp.headers)['Set-Cookie']
     resp = resp.follow()
+    assert "The theme has been changed" in str(resp.html)
     assert resp.form['theme'].value == 'alfortville'
 
     resp.form['theme'].value = 'publik'