correct an error

forget to change before