Fix failing functional test