Fix bug with submit button