fix bug
fix again