minor bug when we use single db
fix minor bug when we use single db connection