Fix type-checking