diff --git a/res/miracle-wfdctl-demo b/res/miracle-wfdctl-demo
index c3bc366..efab5e4 100755
--- a/res/miracle-wfdctl-demo
+++ b/res/miracle-wfdctl-demo
@@ -2,7 +2,6 @@
 PS4="\e[34m> ${0##*/}:\e[0m "
 set -x
 set -e
-trap cleanup EXIT
 
 if (( $# < 1 )); then
     echo Usage: ${0##*/} link-index peer-p2p-mac [nm-link-index]
@@ -145,6 +144,8 @@ peer_disconnect()
 killall miracle-wifid &>/dev/null || true
 killall miracle-wfwctl &>/dev/null || true
 
+trap cleanup EXIT
+
 set_prop org.freedesktop.NetworkManager \
     /org/freedesktop/NetworkManager/Devices/"$nm_link_index" \
     org.freedesktop.NetworkManager.Device \