Forgot to commit fix