Fix build to make it work on WildFly as well
Merge pull request #1 from ctomc/master

Fix build to make it work on WildFly as well