Fix shitty code

Signed-off-by: Charlotte W <cwelkin@icloud.com>