Forgot to update a couple of things