Fixed the StatefulEntitiesCallbacksListener