Fix type inference for function parameters