Fix type error in previous commit