test fix to match new logic