Fix failing test

[prakashchoudhary07]

          > @prakashchoudhary07 the tests are failing here

Will be fixing these in separate PR

Reason: Its because of the checkDb function which has been added, and currently CI don't connect to DB

Originally posted by @prakashchoudhary07 in #93 (comment)