Fix stupid mistake.

Mistakenly removed switch case re-added.