Level editor fixes