UIBehavior: Fix bug