Fixed issue with extension