[IMPROVEMENT] minor fix