Fix plain js wrapping