Fix test helper that was failing due to last commit