diff --git a/src/main/java/org/ntlab/acanthus_server/entities/Work.java b/src/main/java/org/ntlab/acanthus_server/entities/Work.java
index 21c36ef..bdaec55 100644
--- a/src/main/java/org/ntlab/acanthus_server/entities/Work.java
+++ b/src/main/java/org/ntlab/acanthus_server/entities/Work.java
@@ -1,16 +1,19 @@
 package org.ntlab.acanthus_server.entities;
 
+import org.ntlab.acanthus_server.models.Gallery;
+
 public class Work {
 
     int status = 0;
-    Animation animation = null;
+    private Gallery animations = Gallery.getInstance();
 
     public Animation getAnimation(){ return this.animation; }
 
     public void setAnimation(Animation anime) { this.animation = anime; }
 
-    public void setInvites(Animation anime){
-        this.animation = anime;
+    public void setInvites(int aid){
+        Animation animation = animations.getAnimationByAid(aid);
+        this.setAnimation(animation);
         this.status = 1;
     }