Fixed a couple of issue with WP3.8 new UI