added imports for db schema in test
added imports for db schema in test