A few small fixes
small change