diff --git a/.github/workflows/pr_code_changes.yaml b/.github/workflows/pr_code_changes.yaml
index bc2c93975..e63a4d908 100644
--- a/.github/workflows/pr_code_changes.yaml
+++ b/.github/workflows/pr_code_changes.yaml
@@ -50,13 +50,5 @@ jobs:
             run: make test
             env:
               HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
-          - name: Update tests
-            run: make update-tests
-            env:
-              HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
-          - name: Save dataset
-            run: make update-tests
-            env:
-              HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
           - name: Test documentation builds
             run: make documentation
\ No newline at end of file
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
index e69de29bb..fbf2c04bd 100644
--- a/changelog_entry.yaml
+++ b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    fixed:
+    - Bug in loading entity tables.
diff --git a/policyengine_uk/data/dataset_schema.py b/policyengine_uk/data/dataset_schema.py
index f85619264..fdfb9abd1 100644
--- a/policyengine_uk/data/dataset_schema.py
+++ b/policyengine_uk/data/dataset_schema.py
@@ -48,7 +48,7 @@ def __init__(
                 self.person = f["person"]
                 self.benunit = f["benunit"]
                 self.household = f["household"]
-                self.time_period = f["time_period"].iloc[0]
+                self.time_period = str(f["time_period"].iloc[0])
         else:
             if person is None or benunit is None or household is None:
                 raise ValueError(
@@ -57,9 +57,9 @@ def __init__(
             self.person = person
             self.benunit = benunit
             self.household = household
+            self.time_period = str(fiscal_year)
 
         self.data_format = "arrays"
-        self.time_period = fiscal_year
         self.tables = (self.person, self.benunit, self.household)
 
     def save(self, file_path: str):