Actually fixed the test