adjust test, we have added one funtion to the extension