Fix type error

Regression from 4a9e4c3