Forgot some rubbish in the code, fixing it