small fixes

started work on the plugin again :smile: