Fix type oversight