[ revise ] minor bug fix