Fixed up the test