chore(tests): fix up functional tests