sthagen
diff --git a/‎compiler/src/compile.rs‎
Lines changed: 14 additions & 13 deletions b/‎compiler/src/compile.rs‎
Lines changed: 14 additions & 13 deletions
diff --git a/‎tests/snippets/function.py‎
Lines changed: 5 additions & 0 deletions b/‎tests/snippets/function.py‎
Lines changed: 5 additions & 0 deletions
@@ -1015,22 +1015,23 @@ impl<O: OutputStream> Compiler<O> {
     }
 
     fn store_docstring(&mut self, doc_str: Option<String>) {
-        if let Some(doc_string) = doc_str {
-            // Duplicate top of stack (the function or class object)
-            self.emit(Instruction::Duplicate);
+        // Duplicate top of stack (the function or class object)
+        self.emit(Instruction::Duplicate);
 
-            // Doc string value:
-            self.emit(Instruction::LoadConst {
-                value: bytecode::Constant::String {
-                    value: doc_string.to_string(),
+        // Doc string value:
+        self.emit(Instruction::LoadConst {
+            value: match doc_str {
+                Some(doc) => bytecode::Constant::String {
+                    value: doc.to_string(),
                 },
-            });
+                None => bytecode::Constant::None, // set docstring None if not declared
+            },
+        });
 
-            self.emit(Instruction::Rotate { amount: 2 });
-            self.emit(Instruction::StoreAttr {
-                name: "__doc__".to_string(),
-            });
-        }
+        self.emit(Instruction::Rotate { amount: 2 });
+        self.emit(Instruction::StoreAttr {
+            name: "__doc__".to_string(),
+        });
     }
 
     fn compile_while(
 
@@ -95,3 +95,8 @@ def f8() -> int:
 
 with assert_raises(SyntaxError):
     exec('print(keyword=10, 20)')
+
+def f9():
+    pass
+
+assert f9.__doc__ == None