diff --git a/version.php b/version.php
index dc5b736..e37c340 100644
--- a/version.php
+++ b/version.php
@@ -33,13 +33,16 @@ $module->component = 'mod_attendance';
 // TODO: remove this asap.
 if (defined('MOODLE_INTERNAL')) { // Only run if config.php has already been included.
     global $DB;
+    $moduleexists = false;
+
     try {
-        if ($DB->record_exists('modules', array('name' =>'attforblock'))) {
-            require_once('locallib.php');
-            attforblock_upgrade();
-        }
+        $moduleexists = $DB->record_exists('modules', array('name' =>'attforblock'));
     } catch (Exception $e) {
         // Probably a fresh install - modules table doesn't exist
     }
+    if ($moduleexists) {
+        require_once('locallib.php');
+        attforblock_upgrade();
+    }
 
 }