correct the type

correct the type