diff --git a/templates/machine.h.motemplate b/templates/machine.h.motemplate
index 2610cba2..05c5024c 100644
--- a/templates/machine.h.motemplate
+++ b/templates/machine.h.motemplate
@@ -42,7 +42,7 @@ NS_ASSUME_NONNULL_BEGIN
 @interface _<$managedObjectClassName$> : <$customSuperentity$>
 + (instancetype)insertInManagedObjectContext:(NSManagedObjectContext *)moc_;
 + (NSString*)entityName;
-+ (NSEntityDescription*)entityInManagedObjectContext:(NSManagedObjectContext*)moc_;
++ (nullable NSEntityDescription*)entityInManagedObjectContext:(NSManagedObjectContext*)moc_;
 @property (nonatomic, readonly, strong) <$managedObjectClassName$>ID *objectID;
 
 <$foreach Attribute noninheritedAttributes do$>