diff --git a/PKGBUILD b/PKGBUILD
index 99df474..d21d5f8 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,6 +1,6 @@
 # Maintainer: David Thurstenson <thurstylark@gmail.com>
 pkgname=xautolock-systemd
-pkgver=0.1
+pkgver=0.2
 pkgrel=1
 pkgdesc="Systemd unit to run 'xautolock -locknow' before suspend"
 arch=('any')
@@ -8,7 +8,7 @@ url="https://git.thurstylark.com/xautolock-systemd.git/"
 license=('GPL')
 depends=('xautolock')
 source=("xautolock@.service")
-md5sums=('f372d6cce05b70bce2191575d1789654')
+md5sums=('85bb420848607f76c19fd8094b5597eb')
 
 package() {
 	install -Dm644 ${srcdir}/xautolock@.service ${pkgdir}/etc/systemd/system/xautolock@.service
diff --git a/xautolock@.service b/xautolock@.service
index 0513dd9..cd014a3 100644
--- a/xautolock@.service
+++ b/xautolock@.service
@@ -5,7 +5,9 @@ Before=sleep.target
 [Service]
 User=%I
 Type=simple
+Environment=DISPLAY=:0
 ExecStart=/usr/bin/xautolock -locknow
+ExecStartPost=/usr/bin/sleep 1
 
 [Install]
-WantedBy=suspend.target
+WantedBy=sleep.target