diff --git a/test/test_multitensor.py b/test/test_multitensor.py
index 28b118b619..2fe5e2e71f 100644
--- a/test/test_multitensor.py
+++ b/test/test_multitensor.py
@@ -84,7 +84,7 @@ class TestMultiTensor(unittest.TestCase):
     for si, ei in zip(sched[:], lower_schedule(sched)):
       if isinstance(ei.prg, CompiledRunner): names.append(ei.prg.p.name)
       ei.run()
-    assert names[-2] == names[-1], "function was relinearized"
+    self.assertEqual(len(set(names)), 3), "function was relinearized"
 
   @unittest.skip("this doesn't fold because shard_ calls contiguous on all lbs")
   def test_sharded_memory(self):