Fixed something stupid I had done