Fix type expectation