This should be just `when arg` now.