diff --git a/test/test_api/test_classes.py b/test/test_api/test_classes.py
index 080a6d57..0c3dac5a 100644
--- a/test/test_api/test_classes.py
+++ b/test/test_api/test_classes.py
@@ -348,8 +348,8 @@ def test_parent_on_comprehension(Script):
 
 def test_type(Script):
     for c in Script('a = [str()]; a[0].').complete():
-        if c.name == '__class__' and False:  # TODO fix.
-            assert c.type == 'class'
+        if c.name == '__class__':
+            assert c.type == 'property'
         else:
             assert c.type in ('function', 'statement')