fix memory usage test case