Fixing tests.
changing tests