1 files changed, 8 insertions, 11 deletions
diff --git a/vhba-module/vhba-module.install b/vhba-module/vhba-module.install
index 4f7e460..422ff82 100644
--- a/vhba-module/vhba-module.install
+++ b/vhba-module/vhba-module.install
@@ -1,19 +1,16 @@
-# Copied/modified from frandom.install
-
-# arg 1: the new package version
-# arg 2: the old package version
-
-post_install () {
+post_install() {
   echo ">> Place 'vhba' in MODULES= in /etc/rc.conf to enable vhba on system boot."
   echo ">> This module needs to be recompiled for every kernel version upgrade."
-  depmod -a
+  KERNEL_VERSION='2.6.32-ARCH'
+  depmod $KERNEL_VERSION > /dev/null 2>&1
 }
 
-pre_remove () {
-  rmmod vhba 2> /dev/null
+post_upgrade() {
+  post_install
 }
 
-post_remove () {
-  depmod -a
+post_remove() {
+  KERNEL_VERSION='2.6.32-ARCH'
+  depmod $KERNEL_VERSION > /dev/null 2>&1
 }