Fix import inside db/__init__.py