Fixed failing test which was broken by f6117f4