Fix for Travis