fix for in-memory unit tests