diff --git a/MiniC/test_codegen.py b/MiniC/test_codegen.py
index e6a16f1..85802ef 100644
--- a/MiniC/test_codegen.py
+++ b/MiniC/test_codegen.py
@@ -226,7 +226,7 @@ class TestCodeGen(TestExpectPragmas):
         actual = self.all_in_mem(filename, expect)
         self.assert_equal(actual, expect)
 
-    @pytest.mark.parametrize('filename', ALLOC_FILES)
+    @pytest.mark.parametrize('filename', ALL_IN_MEM_FILES)
     def test_smart_alloc(self, filename):
         """Generate code with smart allocation."""
         expect = self.get_expect(filename)