diff --git a/library.json b/library.json
index 97e029a..7f97eaa 100644
--- a/library.json
+++ b/library.json
@@ -1,6 +1,6 @@
 {
     "name": "Wifi",
-    "version": "2.0.1",
+    "version": "2.0.2",
     "description": "Helper classes for handling wifi connectiviy and OTA",
     "repository":
     {
@@ -15,6 +15,9 @@
         "maintainer": true
       }
     ],
+    "dependencies": {
+      "arkhipenko/TaskScheduler": "^3.8.5"
+    },
     "license": "MIT",
     "frameworks": "arduino",
     "platforms": "*"
diff --git a/src/ota.h b/src/ota.h
index def614f..e1731c7 100644
--- a/src/ota.h
+++ b/src/ota.h
@@ -1,4 +1,7 @@
 #include <ArduinoOTA.h>
+#include <TaskScheduler.h>
+
+extern Scheduler ts;
 
 namespace  Ota {
 
diff --git a/src/wifi.h b/src/wifi.h
index 07b92f9..30b09e2 100644
--- a/src/wifi.h
+++ b/src/wifi.h
@@ -1,7 +1,10 @@
 #include <ESP8266WiFi.h>
 #include <ESP8266mDNS.h>
+#include <TaskScheduler.h>
 #include "credentials.h"
 
+extern Scheduler ts;
+
 namespace Wifi {
 
     WiFiEventHandler stationConnectedHandler;