Bug fix for travis