Fix for travis