Fixed the test implementation of DBConnector in main