Modify test example to access the database