fix little type bugs