Fix race condition in editor causing WakeAIZ to try to use BaseEntity.Query.Server before it's set (use InitShared callback instead of Awake)