diff --git a/button.py b/button.py
index 77b7315..618ba3f 100644
--- a/button.py
+++ b/button.py
@@ -1,3 +1,5 @@
+import pygame
+
 class Button():
 	def __init__(self, x, y, image, scale):
 		width = image.get_width()