diff --git a/llm_studio/app_utils/config.py b/llm_studio/app_utils/config.py
index 2e0b4ebe3..77c30d7a4 100644
--- a/llm_studio/app_utils/config.py
+++ b/llm_studio/app_utils/config.py
@@ -20,7 +20,7 @@ def get_size(x):
         return 2**31
 
 
-version = "1.4.0-dev"
+version = "1.3.1"
 
 try:
     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
diff --git a/llm_studio/app_utils/db.py b/llm_studio/app_utils/db.py
index 1967cbf22..bb41e9622 100644
--- a/llm_studio/app_utils/db.py
+++ b/llm_studio/app_utils/db.py
@@ -1,3 +1,5 @@
+from typing import Optional
+
 import pandas as pd
 from pandas.core.frame import DataFrame
 from sqlalchemy import Integer, String, create_engine
@@ -18,7 +20,9 @@ class Dataset(Base):
     path: Mapped[str] = mapped_column("path", String)
     config_file: Mapped[str] = mapped_column("config_file", String)
     train_rows: Mapped[int] = mapped_column("train_rows", Integer)
-    validation_rows: Mapped[int] = mapped_column("validation_rows", Integer)
+    validation_rows: Mapped[Optional[int]] = mapped_column(
+        "validation_rows", Integer, nullable=True
+    )
 
 
 class Experiment(Base):