Fix error
Update test after change the function