fix a bug
Fix 2 bugs