Removed some unnecessary test code from the module