From 9c1f2f6bbe8505abb88162ab09548c5ee6889f56 Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Fri, 5 Jun 2020 17:28:43 -0400
Subject: [PATCH] Update fitbit_parse_sleep.py

---
 src/data/fitbit_parse_sensors/fitbit_parse_sleep.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/data/fitbit_parse_sensors/fitbit_parse_sleep.py b/src/data/fitbit_parse_sensors/fitbit_parse_sleep.py
index a6e3c1c9..8f1450c4 100644
--- a/src/data/fitbit_parse_sensors/fitbit_parse_sleep.py
+++ b/src/data/fitbit_parse_sensors/fitbit_parse_sleep.py
@@ -200,8 +200,9 @@ def parseOneRecordForV12(record, device_id, d_is_main_sleep, records_summary, re
 
 
 def parseSleepData(sleep_data, HOUR2EPOCH):
+    SLEEP_SUMMARY_COLUMNS = SLEEP_SUMMARY_COLUMNS_V1_2
     if sleep_data.empty:
-        return pd.DataFrame(columns=SLEEP_SUMMARY_COLUMNS_V1), pd.DataFrame(columns=SLEEP_INTRADAY_COLUMNS)
+        return pd.DataFrame(columns=SLEEP_SUMMARY_COLUMNS), pd.DataFrame(columns=SLEEP_INTRADAY_COLUMNS)
     device_id = sleep_data["device_id"].iloc[0]
     records_summary, records_intraday = [], []
     # Parse JSON into individual records