Fix type in type hint

This parameter can also be a boolean or an array