diff --git a/epc_data/app.py b/epc_data/app.py
index c6007119..b7ff54bd 100644
--- a/epc_data/app.py
+++ b/epc_data/app.py
@@ -30,16 +30,9 @@ def handler():
                 params={"local-authority": la},
                 page_size=5000,
                 n_pages=10,
-            )["rows"]
+            )
         )
 
-    # TODO: Temp - pull in sample
-    from collections import Counter
-    import pickle
-    from pprint import pprint
-    with open("./epc_data/test_epc_data.obj", "rb") as f:
-        data = pickle.load(f)
-
     # TODO: Fill this
     ClEANING_FIELDS = [
         "roof-description",