Fixed bug for button disapearing on screen size change.