Fixing bug with test helper.