diff --git a/suspend_rtw89 b/suspend_rtw89
index 5f1183e..2d49c51 100755
--- a/suspend_rtw89
+++ b/suspend_rtw89
@@ -1,6 +1,5 @@
 #!/bin/sh
-if [ "${1}" == "pre" ]; then
-  modprobe -rv rtw_8852ae
-elif [ "${1}" == "post" ]; then
-  modprobe -v rtw_8852ae
-fi
+case "$1" in
+pre) modprobe -rv rtw_8852ae ;;
+post) modprobe -v rtw_8852ae ;;
+esac