Fixing error in test module