Fix test runner for Travis