diff --git a/server.py b/server.py
index c4a125f..a1d2417 100644
--- a/server.py
+++ b/server.py
@@ -1,6 +1,7 @@
 # pylint: disable=wrong-import-order
 
 import headscale, helper, json, os, pytz, renderer, secrets, urllib
+from functools      import wraps
 from flask          import Flask, Markup, redirect, render_template, request, url_for, logging
 from dateutil       import parser
 from flask_executor import Executor