Fix broken return value prompt element