Fix type from last commit