diff --git a/src/main/java/cc/fascinated/bat/command/BatCommand.java b/src/main/java/cc/fascinated/bat/command/BatCommand.java
index c4986a2..7884a03 100644
--- a/src/main/java/cc/fascinated/bat/command/BatCommand.java
+++ b/src/main/java/cc/fascinated/bat/command/BatCommand.java
@@ -64,15 +64,10 @@ public abstract class BatCommand {
         if (info.isUserInstall()) {
             integrationTypes.add(IntegrationType.USER_INSTALL);
         }
-        List<InteractionContextType> contextTypes = new ArrayList<>();
-        if (info.isGuildOnly()) {
-            contextTypes.add(InteractionContextType.GUILD);
-        } else {
-            contextTypes.addAll(InteractionContextType.ALL);
-        }
+        System.out.println(integrationTypes);
         commandData = new CommandDataImpl(info.getName(), info.getDescription())
-                .setContexts(contextTypes)
-                .setIntegrationTypes(integrationTypes);
+                .setIntegrationTypes(integrationTypes)
+                .setGuildOnly(info.isGuildOnly());
     }
 
     /**