Works with in memory DB now