Fixed another test that wasnt working