fix bug
FIX BUG