Still trying to fix this test