Add support for both db managers