Fixing tests.
Fixed test.