Fix ee update logic