Fixing travis test