diff --git a/plexpy/notification_handler.py b/plexpy/notification_handler.py
index cf5a15df..62c83e43 100644
--- a/plexpy/notification_handler.py
+++ b/plexpy/notification_handler.py
@@ -269,7 +269,7 @@ def notify_timeline(timeline_data=None, notify_action=None):
                                             metadata=metadata)
 
                 # Set the notification state in the db
-                set_notify_state(session=stream_data,
+                set_notify_state(session=timeline_data,
                                  notify_action=notify_action,
                                  agent_info=agent,
                                  notify_strings=notify_strings,