Fixed issue with tests.