diff --git a/states/_modules/linux_netstat.py b/states/_modules/linux_netstat.py index e78bab4..8b3a35b 100644 --- a/states/_modules/linux_netstat.py +++ b/states/_modules/linux_netstat.py @@ -1,8 +1,10 @@ +__virtualname__= 'netstat' + def __virtual__(): """ Only run on Linux systems """ - return 'netstat' if __grains__['kernel'] == 'Linux' else False + return __virtualname__ if __grains__['kernel'] == 'Linux' else False def s(): """ @@ -26,4 +28,4 @@ def s(): stats[prefix][headers[pos]] = int(items[pos]) currently_in_header_line = not currently_in_header_line - return stats \ No newline at end of file + return stats