Fix dropped batch counting