Add to drop_dims check next value, a little sloppy