1 files changed, 7 insertions, 7 deletions
diff --git a/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java b/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
index 54aa4d1..745e382 100644
--- a/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
+++ b/ufund-api/src/test/java/com/ufund/api/ufundapi/model/UserTest.java
@@ -1,14 +1,14 @@
 package com.ufund.api.ufundapi.model;
 
-import org.junit.jupiter.api.Tag;
-import org.junit.jupiter.api.Test;
-
-import static org.junit.jupiter.api.Assertions.*;
-import static org.mockito.Mockito.mock;
-
 import java.io.IOException;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
 import com.ufund.api.ufundapi.service.CupboardService;
@@ -77,7 +77,7 @@ public class UserTest {
         when(cupboardService.getNeed(0)).thenReturn(need2);
 
         user.addToBasket(need);
-        user.removeBasketNeed(need);
+        user.removeBasketNeed(need.getId());
         user.addToBasket(need2);
 
         Need getNeed = cupboardService.getNeed(user.getBasketNeeds()[0]);