diff --git a/src/com/droidquest/chipstuff/ChipCompiler.java b/src/com/droidquest/chipstuff/ChipCompiler.java
index 50ea4bc..fa059e4 100644
--- a/src/com/droidquest/chipstuff/ChipCompiler.java
+++ b/src/com/droidquest/chipstuff/ChipCompiler.java
@@ -48,22 +48,22 @@ public ChipCompiler(PrototypeChip pc, SmallChip sc)
 		    Device device = (Device) item;
 		    Gate gate=null;
 		    String type = item.getClass().toString();
-		    if (type.endsWith("ANDGate"))
+		    if (device instanceof com.droidquest.devices.ANDGate)
 		      gate = new Gate("AND");
-		    if (type.endsWith("ORGate"))
+		    if (device instanceof com.droidquest.devices.ORGate)
 		      gate = new Gate("OR");
-		    if (type.endsWith("NOTGate"))
+		    if (device instanceof com.droidquest.devices.NOTGate)
 		      gate = new Gate("NOT");
-		    if (type.endsWith("XORGate"))
+		    if (device instanceof com.droidquest.devices.XORGate)
 		      gate = new Gate("XOR");
-		    if (type.endsWith("FlipFlop"))
+		    if (device instanceof com.droidquest.devices.FlipFlop)
 		      {
 			 gate = new Gate("FF");
 			 gate.state = ((FlipFlop)device).state;
 		      }
-		    if (type.endsWith("Node"))
+		    if (device instanceof com.droidquest.devices.Node)
 		      gate = new Gate("NODE");
-		    if (type.endsWith("SmallChip"))
+		    if (device instanceof com.droidquest.devices.SmallChip)
 		      gate = new Gate((SmallChip)device);
 		    if (gate != null)
 		      {