Fix types

Fixed type incompatibility