diff --git a/pyproject.toml b/pyproject.toml
index 9093d1e..280393e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "rialto"
 
-version = "2.0.3"
+version = "2.0.4"
 
 packages = [
     { include = "rialto" },
diff --git a/rialto/runner/bookkeeper.py b/rialto/runner/bookkeeper.py
index a03ff6e..c1718d8 100644
--- a/rialto/runner/bookkeeper.py
+++ b/rialto/runner/bookkeeper.py
@@ -38,7 +38,7 @@ def add(self, record: Record) -> None:
 
         :param record: Record to add to the table.
         """
-        new = self.spark.createDataFrame([record.to_spark_row()], Record.get_schema())
+        new = self.spark.createDataFrame([record.to_spark_row()], record.get_schema())
         db = self._load()
         if db:
             db = db.unionByName(new)