Fix 2 bugs