FIX BUGS

FIX BUGS