Fix for coverage with db