Fix "Update Editor State Bug"
One more fix