diff --git a/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/CircularDotsLoader.kt b/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/CircularDotsLoader.kt
index 2b04c49..8ba2cc2 100644
--- a/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/CircularDotsLoader.kt
+++ b/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/CircularDotsLoader.kt
@@ -1,6 +1,5 @@
 package com.agrawalsuneet.dotsloader.loaders
 
-import android.app.Activity
 import android.content.Context
 import android.graphics.Canvas
 import android.util.AttributeSet
@@ -68,7 +67,9 @@ class CircularDotsLoader : CircularAbstractView {
                     selectedDotPos = 1
                 }
 
-                (context as Activity).runOnUiThread { invalidate() }
+                handler?.let {
+                    it.post { invalidate() }
+                }
             }
         }, 0, animDur.toLong())
     }
diff --git a/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/LinearDotsLoader.kt b/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/LinearDotsLoader.kt
index bef4cc6..5c30475 100644
--- a/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/LinearDotsLoader.kt
+++ b/dotsloader/src/main/java/com/agrawalsuneet/dotsloader/loaders/LinearDotsLoader.kt
@@ -1,6 +1,5 @@
 package com.agrawalsuneet.dotsloader.loaders
 
-import android.app.Activity
 import android.content.Context
 import android.graphics.Canvas
 import android.util.AttributeSet
@@ -119,7 +118,10 @@ class LinearDotsLoader : DotsLoaderBaseView {
                     }
                 }
 
-                (context as Activity).runOnUiThread { invalidate() }
+                handler?.let {
+                    it.post { invalidate() }
+                }
+
             }
         }, 0, animDur.toLong())
     }