fix the types