Fixed test file
Working on fixing the test