Fixing compile error for test class.