diff --git a/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/PushCommand.java b/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/PushCommand.java index 29f150a6a2..d97df8d569 100644 --- a/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/PushCommand.java +++ b/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/PushCommand.java @@ -18,6 +18,7 @@ import com.denizenscript.denizencore.scripts.ScriptEntry; import com.denizenscript.denizencore.scripts.commands.AbstractCommand; import com.denizenscript.denizencore.scripts.commands.Holdable; +import com.denizenscript.denizencore.scripts.containers.core.TaskScriptContainer; import com.denizenscript.denizencore.scripts.queues.ScriptQueue; import com.denizenscript.denizencore.utilities.ScriptUtilities; import org.bukkit.Location; @@ -118,7 +119,7 @@ else if (!scriptEntry.hasObject("speed") scriptEntry.addObject("speed", arg.asElement()); } else if (!scriptEntry.hasObject("script") - && (arg.matchesArgumentType(ScriptTag.class) + && ((arg.matchesArgumentType(ScriptTag.class) && arg.asType(ScriptTag.class).getContainer() instanceof TaskScriptContainer) || arg.matchesPrefix("script"))) { scriptEntry.addObject("script", arg.asType(ScriptTag.class)); } diff --git a/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/ShootCommand.java b/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/ShootCommand.java index a5fa08d6eb..139ba63f23 100644 --- a/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/ShootCommand.java +++ b/plugin/src/main/java/com/denizenscript/denizen/scripts/commands/entity/ShootCommand.java @@ -17,6 +17,7 @@ import com.denizenscript.denizencore.scripts.ScriptEntry; import com.denizenscript.denizencore.scripts.commands.AbstractCommand; import com.denizenscript.denizencore.scripts.commands.Holdable; +import com.denizenscript.denizencore.scripts.containers.core.TaskScriptContainer; import com.denizenscript.denizencore.scripts.queues.ScriptQueue; import com.denizenscript.denizencore.utilities.CoreUtilities; import com.denizenscript.denizencore.utilities.ScriptUtilities; @@ -135,7 +136,7 @@ else if (!scriptEntry.hasObject("speed") scriptEntry.addObject("speed", arg.asElement()); } else if (!scriptEntry.hasObject("script") - && (arg.matchesArgumentType(ScriptTag.class) + && ((arg.matchesArgumentType(ScriptTag.class) && arg.asType(ScriptTag.class).getContainer() instanceof TaskScriptContainer) || arg.matchesPrefix("script"))) { scriptEntry.addObject("script", arg.asType(ScriptTag.class)); }