Fixed travis test failing