From 9d7c791033c61b3884f9712d24c30c2d638ec525 Mon Sep 17 00:00:00 2001
From: Joe Ksiazek <jak3122@rit.edu>
Date: Thu, 15 Mar 2018 19:20:30 -0400
Subject: [PATCH] Set .sockets to event callbacks

---
 src/Main.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/Main.js b/src/Main.js
index 783eb9c..072a213 100755
--- a/src/Main.js
+++ b/src/Main.js
@@ -19,12 +19,14 @@ export default {
                 Object.defineProperty(this.$options.sockets, 'on', {
                     value: (label, callback) => {
                         Emitter.addListener(label, callback, this);
+                        this.$options.sockets[label] = callback;
                     }
                 });
 
                 Object.defineProperty(this.$options.sockets, 'off', {
                     value: (label, callback) => {
                         Emitter.removeListener(label, callback, this);
+                        delete this.$options.sockets[label];
                     }
                 });
 
@@ -40,6 +42,7 @@ export default {
                 if(sockets){
                     Object.keys(sockets).forEach((key) => {
                         Emitter.removeListener(key, sockets[key], this);
+                        delete this.$options.sockets[key];
                     });
                 }
             }