Fixed brian's shitty code