diff --git a/01_data_sampling_example.ipynb b/01_data_sampling_example.ipynb
index 75d1b65..2d6dd3e 100644
--- a/01_data_sampling_example.ipynb
+++ b/01_data_sampling_example.ipynb
@@ -161,6 +161,7 @@
    "source": [
     "# Only download 101 buildings for sampling purposes and testing this notebook/repo\n",
     "timeseries_to_download = df_sub.head(10)\n",
+    "display(timeseries_to_download)\n",
     "print(f\"Downloading time series data for {len(timeseries_to_download)} buildings into {time_series_data_path}\")\n",
     "\n",
     "time_series_data_file_paths, building_ids = processor.process_building_time_series(\n",
diff --git a/comstock_processor.py b/comstock_processor.py
index 77352cc..24ae680 100644
--- a/comstock_processor.py
+++ b/comstock_processor.py
@@ -117,7 +117,7 @@ def download_task(row):
                 return save_path, building_id
 
             building_time_series_file = (
-                f"{self.time_series_url}/by_state/upgrade={self.upgrade}/state={self.state}/{building_id}-{self.upgrade}.parquet"
+                f"{self.time_series_url}/by_state/upgrade={self.upgrade}/state={row['in.state']}/{building_id}-{self.upgrade}.parquet"
             )
             self.download_file(building_time_series_file, save_path)
             return save_path, building_id
diff --git a/poetry.lock b/poetry.lock
index 009d4ca..3f3ebf7 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -3810,14 +3810,14 @@ zstd = ["zstandard (>=0.18.0)"]
 
 [[package]]
 name = "virtualenv"
-version = "20.35.3"
+version = "20.35.4"
 description = "Virtual Python Environment builder"
 optional = false
 python-versions = ">=3.8"
 groups = ["dev"]
 files = [
-    {file = "virtualenv-20.35.3-py3-none-any.whl", hash = "sha256:63d106565078d8c8d0b206d48080f938a8b25361e19432d2c9db40d2899c810a"},
-    {file = "virtualenv-20.35.3.tar.gz", hash = "sha256:4f1a845d131133bdff10590489610c98c168ff99dc75d6c96853801f7f67af44"},
+    {file = "virtualenv-20.35.4-py3-none-any.whl", hash = "sha256:c21c9cede36c9753eeade68ba7d523529f228a403463376cf821eaae2b650f1b"},
+    {file = "virtualenv-20.35.4.tar.gz", hash = "sha256:643d3914d73d3eeb0c552cbb12d7e82adf0e504dbf86a3182f8771a153a1971c"},
 ]
 
 [package.dependencies]