Fixed shitty code