1 files changed, 5 insertions, 3 deletions
diff --git a/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/common/layout/TestAttributeInfo.java b/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/common/layout/TestAttributeInfo.java
index ee04260..f4f83c1 100644
--- a/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/common/layout/TestAttributeInfo.java
+++ b/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/common/layout/TestAttributeInfo.java
@@ -17,10 +17,12 @@ package com.android.ide.common.layout;
 
 import com.android.ide.common.api.IAttributeInfo;
 
+import java.util.EnumSet;
+
 /** Test/mock implementation of {@link IAttributeInfo} */
 public class TestAttributeInfo implements IAttributeInfo {
     private final String mName;
-    private final Format[] mFormats;
+    private final EnumSet<Format> mFormats;
     private final String mDefinedBy;
     private final String[] mEnumValues;
     private final String[] mFlagValues;
@@ -30,7 +32,7 @@ public class TestAttributeInfo implements IAttributeInfo {
         this(name, null, null, null, null, null);
     }
 
-    public TestAttributeInfo(String name, Format[] formats, String definedBy,
+    public TestAttributeInfo(String name, EnumSet<Format> formats, String definedBy,
             String[] enumValues, String[] flagValues, String javadoc) {
         super();
         this.mName = name;
@@ -57,7 +59,7 @@ public class TestAttributeInfo implements IAttributeInfo {
     }
 
     @Override
-    public Format[] getFormats() {
+    public EnumSet<Format> getFormats() {
         return mFormats;
     }