Fixed a bug with the action button