Fix missing controller