Fix missing using in functional tests