Fix for travis test error