Add missing controller
Fix a bug in the switching mockup