Fix test to run in db transaction
Merge pull request #1 from blowmage/fix

Fix test to run in db transaction