Fix return to flight plan test