Fix reservoir sampling