fix bug
fix 2