updating and fixing tests
changing test to check the fix