diff --git a/generate_config.py b/generate_config.py
index 2af17d7..c99c964 100755
--- a/generate_config.py
+++ b/generate_config.py
@@ -2,6 +2,11 @@
 import binascii
 import yaml
 import os
+import platform
+
+if "Ubuntu" in platform.dist()[0]:
+	if "16.04" in platfrom.dist()[1]:
+		os.system("sudo apt install python-yaml")
 
 nginx_template = """
 server {