Fix type inference test