diff --git a/build.gradle b/build.gradle
index 69192e9..de19a70 100644
--- a/build.gradle
+++ b/build.gradle
@@ -5,7 +5,7 @@ buildscript {
         jcenter()
     }
     dependencies {
-        classpath 'com.android.tools.build:gradle:2.2.2'
+        classpath 'com.android.tools.build:gradle:2.3.3'
         classpath 'com.github.dcendents:android-maven-gradle-plugin:1.5'
 
         // NOTE: Do not place your application dependencies here; they belong
diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties
index 1ffaa19..a19e93b 100644
--- a/gradle/wrapper/gradle-wrapper.properties
+++ b/gradle/wrapper/gradle-wrapper.properties
@@ -1,6 +1,6 @@
-#Mon Dec 05 22:51:00 CST 2016
+#Tue Oct 17 08:42:17 BST 2017
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-2.14.1-all.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-3.3-all.zip
diff --git a/multistatelayout/src/main/java/cn/refactor/multistatelayout/MultiStateLayout.java b/multistatelayout/src/main/java/cn/refactor/multistatelayout/MultiStateLayout.java
index 2d4fa92..cbf84b8 100644
--- a/multistatelayout/src/main/java/cn/refactor/multistatelayout/MultiStateLayout.java
+++ b/multistatelayout/src/main/java/cn/refactor/multistatelayout/MultiStateLayout.java
@@ -218,6 +218,15 @@ public int getState() {
         return mIsSystemState ? mCurState : mCurCustomStateKey;
     }
 
+    /**
+     * check if the current state equals state passed in
+     * @param state
+     * @return
+     */
+    public boolean isState(int state) {
+        return getState() == state;
+    }
+
     /**
      * @return current state is customise state
      */
diff --git a/sample/src/main/java/cn/refactor/multistatelayoutdemo/MainActivity.java b/sample/src/main/java/cn/refactor/multistatelayoutdemo/MainActivity.java
index b82145d..1ec0e8a 100644
--- a/sample/src/main/java/cn/refactor/multistatelayoutdemo/MainActivity.java
+++ b/sample/src/main/java/cn/refactor/multistatelayoutdemo/MainActivity.java
@@ -95,6 +95,10 @@ public boolean onMenuItemClick(MenuItem item) {
                         break;
                 }
                 Log.d(TAG, String.valueOf(mStateLayout.getState()));
+                //check state matches specified value
+                if (mStateLayout.isState(MultiStateLayout.State.ERROR)) {
+                    Toast.makeText(MainActivity.this, "State is now ERROR", Toast.LENGTH_SHORT).show();
+                }
                 return true;
             }
         });