diff --git a/listen-for-shutdown.py b/listen-for-shutdown.py
index cecc42b..3efa4e1 100755
--- a/listen-for-shutdown.py
+++ b/listen-for-shutdown.py
@@ -1,12 +1,11 @@
 #!/usr/bin/env python
 
 
-import RPi.GPIO as GPIO
+from gpiozero import Button
 import subprocess
 
+button = Button(3)
 
-GPIO.setmode(GPIO.BCM)
-GPIO.setup(3, GPIO.IN, pull_up_down=GPIO.PUD_UP)
-GPIO.wait_for_edge(3, GPIO.FALLING)
+button.wait_for_press()
 
 subprocess.call(['shutdown', '-h', 'now'], shell=False)