fixing the bug of the last batch smaller than X