Fixed the test
Working on the test