Correct NonNull annotation

[yangtools.git] / yang / yang-model-api / src / main / java / org / opendaylight / yangtools / yang / model / api / stmt / WhenStatement.java

diff --git a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/WhenStatement.java b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/WhenStatement.java
index ac8d6fffa50096aafb42e5c949124027b3d47c5a..c17a58f8b693c80fbadad65d8d5b3afe9188ef24 100644 (file)
--- a/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/WhenStatement.java
+++ b/yang/yang-model-api/src/main/java/org/opendaylight/yangtools/yang/model/api/stmt/WhenStatement.java
@@ -10,10 +10,17 @@ package org.opendaylight.yangtools.yang.model.api.stmt;
 import static com.google.common.base.Verify.verifyNotNull;
 
 import org.eclipse.jdt.annotation.NonNull;
-import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
+import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
+import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
+import org.opendaylight.yangtools.yang.xpath.api.YangXPathExpression.QualifiedBound;
 
-public interface WhenStatement extends DocumentedDeclaredStatement<RevisionAwareXPath> {
-    default @NonNull RevisionAwareXPath getCondition() {
+public interface WhenStatement extends DocumentedDeclaredStatement<QualifiedBound> {
+    @Override
+    default StatementDefinition statementDefinition() {
+        return YangStmtMapping.WHEN;
+    }
+
+    default @NonNull QualifiedBound getCondition() {
         // FIXME: YANGTOOLS-908: verifyNotNull() should not be needed here
         return verifyNotNull(argument());
     }