Updating test written as part of #362