fixed errors with tests: