Fixes crash with button issue