Should fix travis test