Few improvements in InMemory database registration