Fixed my shitty code.