fix bug
fix bug #2