1 files changed, 2 insertions, 2 deletions
diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/model/User.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/User.java
index d697057..13000cc 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/model/User.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/User.java
@@ -55,8 +55,8 @@ public class User {
         return basket.toArray(Integer[]::new);
     }
 
-    public void removeBasketNeed(Need need) {
-        basket.remove(need.getId());
+    public boolean removeBasketNeed(Integer needID) {
+        return basket.remove(needID);
     }
 
     public User withoutPasswordHash() {