fix button return