Test fixed for Travis