diff --git a/app/src/main/java/model/UniformColorGenerator.java b/app/src/main/java/model/UniformColorGenerator.java
index 9f21c99041f4f8dedfa3aab5bfc1b5eba587f739..6f31a04b4f14c083045445f3ff8ebbde532a58fa 100644
--- a/app/src/main/java/model/UniformColorGenerator.java
+++ b/app/src/main/java/model/UniformColorGenerator.java
@@ -2,18 +2,18 @@ package model;
 
 import javafx.scene.paint.Color;
 
-import java.util.Random;
-
 public class UniformColorGenerator implements ColorGenerator{
 
     private Color color;
 
 
+    public UniformColorGenerator(Color color){
+        this.color = color;
+    }
     @Override
-   public Color nextColor(Cell cells) {
-
-        return this.color;
+    public Color nextColor(Cell cell) {
 
+        return  this.color;
     }
    
 }