Updated unit test for database