Quick change to make button appear properly