Fix 2 small mistakes