Fix minor oops