Update WGIntegration.java

Cleanup wg integration