diff --git a/src/com/android/musicfx/ActivityMusic.java b/src/com/android/musicfx/ActivityMusic.java
index d275a85..68b0001 100644
--- a/src/com/android/musicfx/ActivityMusic.java
+++ b/src/com/android/musicfx/ActivityMusic.java
@@ -251,7 +251,7 @@
             if (deviceInfo == null) {
                 continue;
             }
-            final int type = deviceInfo.getType();
+            final @AudioDeviceInfo.AudioDeviceType int type = deviceInfo.getType();
             if (HEADSET_DEVICE_TYPES.contains(type)) {
                 Log.v(TAG, " at least a HeadSet device type " + type + " connected");
                 return true;