Fixing compilation error in test class.