Actually it works pretty much out of the box with the latest Wildfly Alpha release. I will apply this after the 5.2.1 release.