Fixed up some editor stuff