diff --git a/templates/machine.h.motemplate b/templates/machine.h.motemplate
index 575f0884..ff043d61 100644
--- a/templates/machine.h.motemplate
+++ b/templates/machine.h.motemplate
@@ -32,7 +32,7 @@ extern const struct <$managedObjectClassName$>FetchedProperties {<$foreach Fetch
 <$endforeach do$>
 <$else$>@class <$Attribute.objectAttributeClassName$>;<$endif$><$endif$>
 <$endforeach do$>
-@interface <$managedObjectClassName$>ID : NSManagedObjectID {}
+@interface <$managedObjectClassName$>ID : <$customSuperentity$>ID {}
 @end
 
 @interface _<$managedObjectClassName$> : <$customSuperentity$> {}