Forgot to commit this quick fix