Forgot to remove this piece of code