Add more memory usage in test