diff --git a/src/context.cpp b/src/context.cpp
index 1bfeba2..3b50c70 100644
--- a/src/context.cpp
+++ b/src/context.cpp
@@ -207,8 +207,8 @@ namespace fslic {
 
         // safeguard
         for (int k = 0; k < K; k++) {
-            clusters[k].x = clamp<DistType>(clusters[k].x, 0, W - 1);
-            clusters[k].y = clamp<DistType>(clusters[k].y, 0, H - 1);
+            clusters[k].x = clamp<float>(clusters[k].x, 0, W - 1);
+            clusters[k].y = clamp<float>(clusters[k].y, 0, H - 1);
         }
 
         int T = 2 * S + 32;