Repaired a minor bug on main prompt