Put db in unit tests