Update test_plugin.py