Fix for travis ci