Add controller guard