Support more general BaseEntitySystem rather than only EntitySystem.

Namek · Dec 18, 2015 · 9a5a36b · 9a5a36b
1 parent ee80e36
commit 9a5a36b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/artemis-entity-tracker/src/main/java/net/namekdev/entity_tracker/EntityTracker.java b/artemis-entity-tracker/src/main/java/net/namekdev/entity_tracker/EntityTracker.java
@@ -17,6 +17,7 @@
 import net.namekdev.entity_tracker.utils.serialization.NetworkSerialization;
 
 import com.artemis.Aspect;
+import com.artemis.BaseEntitySystem;
 import com.artemis.BaseSystem;
 import com.artemis.Component;
 import com.artemis.ComponentManager;
@@ -26,7 +27,6 @@
 import com.artemis.Entity;
 import com.artemis.EntitySubscription;
 import com.artemis.EntitySubscription.SubscriptionListener;
-import com.artemis.EntitySystem;
 import com.artemis.Manager;
 import com.artemis.utils.Bag;
 import com.artemis.utils.ImmutableBag;
@@ -99,8 +99,8 @@ private void find42UnicornManagers() {
 			BitSet actives = null;
 			EntitySubscription subscription = null;
 
-			if (system instanceof EntitySystem) {
-				EntitySystem entitySystem = (EntitySystem) system;
+			if (system instanceof BaseEntitySystem) {
+				BaseEntitySystem entitySystem = (BaseEntitySystem) system;
 
 				subscription = entitySystem.getSubscription();
 				aspect = subscription.getAspect();