Fix type for trait