Fixed the test2