Fixed tests for travis ci