fixed test for Travis CI