Fixed issue with plugin