minor fix bug