minor fix
fix bug