Fix issue with db module