Fix for travis test