Forgot to add some hooks