If we move the button logic in helper then we don't need this checks here in controller