updated for latest plugin changes

Just updated to the latest changes, minor stuff.