forgot to fix this test too