Fixed bug out of screen (i think)