Fix missing controller methods in unit test