Fix error in plugin.py