Fixing test
Fixed tests.