Fix return button