Fix for travis CI build