Fix counting test