Fix up the in-memory test