Fix get_memory_usage