diff --git a/lavat.c b/lavat.c
index 7643e3e..23a00bc 100644
--- a/lavat.c
+++ b/lavat.c
@@ -49,9 +49,9 @@ int main(int argc, char *argv[]) {
 
   char *custom2 = custom;
 
-  // if (strlen(custom) > 1 && rim) {
-  //   custom2 = custom + 1;
-  // }
+  if (custom && strlen(custom) > 1 && rim) {
+    custom2 = custom + 1;
+  }
 
   for (int i = 0; i < nballs; i++) {
     balls[i].x = rand() % maxX;