Some improvements for test_mem: