Correct plugin code