diff --git a/implementation/legacy/src/main/java/org/codemc/worldguardwrapper/implementation/legacy/WorldGuardImplementation.java b/implementation/legacy/src/main/java/org/codemc/worldguardwrapper/implementation/legacy/WorldGuardImplementation.java index ed6ac18..38c50e9 100644 --- a/implementation/legacy/src/main/java/org/codemc/worldguardwrapper/implementation/legacy/WorldGuardImplementation.java +++ b/implementation/legacy/src/main/java/org/codemc/worldguardwrapper/implementation/legacy/WorldGuardImplementation.java @@ -106,7 +106,7 @@ public class WorldGuardImplementation implements IWorldGuardImplementation { @Override public Optional queryFlag(Player player, Location location, IWrappedFlag flag) { AbstractWrappedFlag wrappedFlag = (AbstractWrappedFlag) flag; - return wrappedFlag.fromWGValue(queryValue(player, location, wrappedFlag.getHandle())); + return queryValue(player, location, wrappedFlag.getHandle()).flatMap(value -> wrappedFlag.fromWGValue(value)); } @Override diff --git a/implementation/v6/src/main/java/org/codemc/worldguardwrapper/implementation/v6/WorldGuardImplementation.java b/implementation/v6/src/main/java/org/codemc/worldguardwrapper/implementation/v6/WorldGuardImplementation.java index 2852a71..004e53d 100644 --- a/implementation/v6/src/main/java/org/codemc/worldguardwrapper/implementation/v6/WorldGuardImplementation.java +++ b/implementation/v6/src/main/java/org/codemc/worldguardwrapper/implementation/v6/WorldGuardImplementation.java @@ -103,7 +103,7 @@ public class WorldGuardImplementation implements IWorldGuardImplementation { @Override public Optional queryFlag(Player player, Location location, IWrappedFlag flag) { AbstractWrappedFlag wrappedFlag = (AbstractWrappedFlag) flag; - return wrappedFlag.fromWGValue(queryValue(player, location, wrappedFlag.getHandle())); + return queryValue(player, location, wrappedFlag.getHandle()).flatMap(value -> wrappedFlag.fromWGValue(value)); } @SuppressWarnings("unchecked") diff --git a/implementation/v7/src/main/java/org/codemc/worldguardwrapper/implementation/v7/WorldGuardImplementation.java b/implementation/v7/src/main/java/org/codemc/worldguardwrapper/implementation/v7/WorldGuardImplementation.java index 1c0d9f2..2218415 100644 --- a/implementation/v7/src/main/java/org/codemc/worldguardwrapper/implementation/v7/WorldGuardImplementation.java +++ b/implementation/v7/src/main/java/org/codemc/worldguardwrapper/implementation/v7/WorldGuardImplementation.java @@ -107,7 +107,7 @@ public class WorldGuardImplementation implements IWorldGuardImplementation { @Override public Optional queryFlag(Player player, Location location, IWrappedFlag flag) { AbstractWrappedFlag wrappedFlag = (AbstractWrappedFlag) flag; - return wrappedFlag.fromWGValue(queryValue(player, location, wrappedFlag.getHandle())); + return queryValue(player, location, wrappedFlag.getHandle()).flatMap(value -> wrappedFlag.fromWGValue(value)); } @SuppressWarnings("unchecked") diff --git a/implementation/v7fawe/src/main/java/org/codemc/worldguardwrapper/implementation/v7fawe/WorldGuardImplementation.java b/implementation/v7fawe/src/main/java/org/codemc/worldguardwrapper/implementation/v7fawe/WorldGuardImplementation.java index d67d91d..1dd55ec 100644 --- a/implementation/v7fawe/src/main/java/org/codemc/worldguardwrapper/implementation/v7fawe/WorldGuardImplementation.java +++ b/implementation/v7fawe/src/main/java/org/codemc/worldguardwrapper/implementation/v7fawe/WorldGuardImplementation.java @@ -112,7 +112,7 @@ public class WorldGuardImplementation implements IWorldGuardImplementation { @Override public Optional queryFlag(Player player, Location location, IWrappedFlag flag) { AbstractWrappedFlag wrappedFlag = (AbstractWrappedFlag) flag; - return wrappedFlag.fromWGValue(queryValue(player, location, wrappedFlag.getHandle())); + return queryValue(player, location, wrappedFlag.getHandle()).flatMap(value -> wrappedFlag.fromWGValue(value)); } @SuppressWarnings("unchecked")