Fixed tests.
Fixed tests.