Removed database stuff as its not needed for this test