diff --git a/docs/concepts/models/python_models.md b/docs/concepts/models/python_models.md
index a66a5649f4..a5380ef6f6 100644
--- a/docs/concepts/models/python_models.md
+++ b/docs/concepts/models/python_models.md
@@ -302,6 +302,7 @@ import typing as t
 from datetime import datetime
 
 import pandas as pd
+from sqlglot.expressions import to_column
 from sqlmesh import ExecutionContext, model
 
 @model(
@@ -317,7 +318,7 @@ from sqlmesh import ExecutionContext, model
         "name": "Name corresponding to the ID",
     },
     audits=[
-        ("not_null", {"columns": ["id"]}),
+        ("not_null", {"columns": [to_column("id")]}),
     ],
 )
 def execute(