Small bug fix to multi-step form