Updated the extension syntax due to migration issues.