Cleaning up JSS code

2 files changed, 2 insertions, 4 deletions

diff --git a/javaparser-testing/src/test/java/com/github/javaparser/JavaTokenTest.java b/javaparser-testing/src/test/java/com/github/javaparser/JavaTokenTest.java
index 534d94349..e69a76ab0 100644
--- a/javaparser-testing/src/test/java/com/github/javaparser/JavaTokenTest.java
+++ b/javaparser-testing/src/test/java/com/github/javaparser/JavaTokenTest.java
@@ -25,7 +25,6 @@ import com.github.javaparser.ast.expr.Expression;
 import org.junit.Test;
 
 import java.util.Iterator;
-import java.util.List;
 
 import static com.github.javaparser.GeneratedJavaParserConstants.*;
 import static com.github.javaparser.JavaToken.Category.*;
@@ -39,8 +38,7 @@ public class JavaTokenTest {
     @Test
     public void testAFewTokens() {
         ParseResult<Expression> result = new JavaParser().parse(ParseStart.EXPRESSION, provider("1 +/*2*/1 "));
-        List<JavaToken> tokens = result.getTokens().get();
-        Iterator<JavaToken> iterator = tokens.iterator();
+        Iterator<JavaToken> iterator = result.getResult().get().getTokenRange().get().iterator();
         assertToken("1", range(1, 1, 1, 1), INTEGER_LITERAL, LITERAL, iterator.next());
         assertToken(" ", range(1, 2, 1, 2), SPACE, WHITESPACE_NO_EOL, iterator.next());
         assertToken("+", range(1, 3, 1, 3), PLUS, OPERATOR, iterator.next());
diff --git a/javaparser-testing/src/test/java/com/github/javaparser/ParserConfigurationTest.java b/javaparser-testing/src/test/java/com/github/javaparser/ParserConfigurationTest.java
index ab4bb7498..ab018ccf1 100644
--- a/javaparser-testing/src/test/java/com/github/javaparser/ParserConfigurationTest.java
+++ b/javaparser-testing/src/test/java/com/github/javaparser/ParserConfigurationTest.java
@@ -13,7 +13,7 @@ public class ParserConfigurationTest {
     public void storeNoTokens() {
         ParseResult<CompilationUnit> result = new JavaParser(new ParserConfiguration().setStoreTokens(false)).parse(ParseStart.COMPILATION_UNIT, provider("class X{}"));
         
-        assertFalse(result.getTokens().isPresent());
+        assertFalse(result.getResult().get().getTokenRange().isPresent());
         assertTrue(result.getResult().get().findAll(Node.class).stream().noneMatch(node -> node.getTokenRange().isPresent()));
     }
 }