Updated module to be compatible with Silverstripe 3.1.x