Cleaning up memory tests