diff --git a/check_opnsense.py b/check_opnsense.py
index 2d2cee7..cef38ee 100755
--- a/check_opnsense.py
+++ b/check_opnsense.py
@@ -5,7 +5,7 @@
 # check_opnsense.py - A check plugin for monitoring OPNsense firewalls.
 # Copyright (C) 2018  Nicolai Buchwitz <nb@tipi-net.de>
 #
-# Version: 0.1.0
+# Version: 0.1.1
 #
 # ------------------------------------------------------------------------------
 # This program is free software; you can redistribute it and/or
@@ -161,13 +161,12 @@ def checkUpdates(self):
         url = self.getURL('core/firmware/status')
         data = self.request(url)
 
-        if data['status'] == 'ok' and data['status_upgrade_action'] == 'all':
-            count = data['updates']
+        if data['status'] == 'update':
 
             self.checkResult = NagiosState.WARNING
-            self.checkMessage = "{} pending updates".format(count)
+            self.checkMessage = data['status_msg']
 
-            if data['upgrade_needs_reboot']:
+            if data['needs_reboot']:
                 self.checkResult = NagiosState.CRITICAL
                 self.checkMessage = "{}. Subsequent reboot required.".format(self.checkMessage)
         else:
@@ -179,4 +178,3 @@ def __init__(self):
 
 opnsense = CheckOPNsense()
 opnsense.check()
-