diff --git a/endpoints/ip_info.py b/endpoints/ip_info.py
index f32ee2c..cefb947 100644
--- a/endpoints/ip_info.py
+++ b/endpoints/ip_info.py
@@ -1,5 +1,6 @@
 from flask_restful import Resource
 import psutil
+import socket
 
 class Ip_Info(Resource):
     def get_ip_addresses(self, family):
diff --git a/endpoints/memory.py b/endpoints/memory.py
index e68c9fd..3b00d7c 100644
--- a/endpoints/memory.py
+++ b/endpoints/memory.py
@@ -1,4 +1,5 @@
 from flask_restful import Resource, abort
+import psutil
 
 class Memory(Resource):
     def get(self):