Working on fixing the test