Fixed my bug but it still looks shitty