fix level editor to work with new refactor