correct the types