Minor type fix