Fix UI controls