Fix reference to old DB class