diff --git a/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java b/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java
index 9776657..4f12f89 100644
--- a/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java
+++ b/src/main/java/net/echo/brain4j/model/impl/FeedForwardModel.java
@@ -30,7 +30,6 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
-import java.util.concurrent.atomic.AtomicReference;
 
 public class FeedForwardModel implements Model {
 
diff --git a/src/main/java/net/echo/brain4j/training/BackPropagation.java b/src/main/java/net/echo/brain4j/training/BackPropagation.java
index fff1a29..3f48fd7 100644
--- a/src/main/java/net/echo/brain4j/training/BackPropagation.java
+++ b/src/main/java/net/echo/brain4j/training/BackPropagation.java
@@ -9,7 +9,7 @@
 import net.echo.brain4j.training.data.DataSet;
 import net.echo.brain4j.training.optimizers.Optimizer;
 
-import java.util.*;
+import java.util.List;
 
 public class BackPropagation {