Fix possability of some editor UI components not being available sometimes. Closes issue #2.