JAL-1601 Expand secondary structure prediction tests

[jalview.git] / src / jalview / datamodel / AllColsIterator.java

diff --git a/src/jalview/datamodel/AllColsIterator.java b/src/jalview/datamodel/AllColsIterator.java
index 67d7ba5..1ff6a66 100644 (file)
--- a/src/jalview/datamodel/AllColsIterator.java
+++ b/src/jalview/datamodel/AllColsIterator.java
@@ -39,7 +39,7 @@ public class AllColsIterator implements Iterator<Integer>
   private int current;
 
   public AllColsIterator(int firstcol, int lastcol,
-          ColumnSelection hiddenCols)
+          HiddenColumns hiddenCols)
   {
     last = lastcol;
     next = firstcol;
@@ -49,13 +49,13 @@ public class AllColsIterator implements Iterator<Integer>
   @Override
   public boolean hasNext()
   {
-    return current + 1 <= last;
+    return next <= last;
   }
 
   @Override
   public Integer next()
   {
-    if (current + 1 > last)
+    if (next > last)
     {
       throw new NoSuchElementException();
     }
@@ -71,4 +71,3 @@ public class AllColsIterator implements Iterator<Integer>
     throw new UnsupportedOperationException();
   }
 }
-