minor updates

2019-04-18 16:12:54 +10:00
parent e6c397397b
commit 2b6afe31c2
3 changed files with 4 additions and 2 deletions
--- a/vulnwhisp/frameworks/nessus.py
+++ b/vulnwhisp/frameworks/nessus.py
@ -227,7 +227,7 @@ class NessusAPI(object):

        # Map risk to a SEVERITY MAPPING value
        self.logger.debug('Mapping risk to severity number')
-        df['risk_number'] = df['risk'].str.lower().map(self.SEVERITY_MAPPING)
+        df['risk_number'] = df['risk'].map(self.SEVERITY_MAPPING)

        df.fillna('', inplace=True)