fixed return button error