fix memory usage test