diff --git a/include/mruby/proc.h b/include/mruby/proc.h
index 9d197f48a..bd1b5a705 100644
--- a/include/mruby/proc.h
+++ b/include/mruby/proc.h
@@ -157,13 +157,11 @@ static inline void
 mrb_vm_ci_target_class_set(mrb_callinfo *ci, struct RClass *tc)
 {
   struct REnv *e = ci->u.env;
-  if (e) {
-    if (e->tt == MRB_TT_ENV) {
-      e->c = tc;
-    }
-    else {
-      ci->u.target_class = tc;
-    }
+  if (e && e->tt == MRB_TT_ENV) {
+    e->c = tc;
+  }
+  else {
+    ci->u.target_class = tc;
   }
 }