Fixed the test