small refactoring

better not have sad faces in the plugin ;)