Fix my broken code :(