Fix build to make it work on WildFly as well