Add additional test for UI Controller