Fix bug in return button.