fixed one more test

Other ones will be harder