Fix test failure due to function signature change
Fix test failure due to function signature change

Mellow-Programming-Language/Mellow,Mellow-Programming-Language/Mellow,Mellow-Programming-Language/Mellow