diff --git a/check_modules.sh b/check_modules.sh
index 91fe10a..46b7a9d 100755
--- a/check_modules.sh
+++ b/check_modules.sh
@@ -13,9 +13,11 @@ logfile="check_modules.log"
 module_load_logfile="$TMP/.module_load.log"
 module_rm_logfile="$TMP/.module_rm.log"
 
+moduleclass=$1
+
 echo "Module environment check, $(date)" | tee $logfile
 
-for m in $(module av -t 2>&1); do
+for m in $(module av -t $moduleclass 2>&1); do
     if [[ $m =~ ^[A-Za-z] ]]; then
         m=$(echo $m | sed -e 's/(.*)//')
         echo -n "Checking $m ... " | tee -a $logfile