Revert "add function argument in test"

This reverts commit fa103e50653936dde631eeead43781593f75f2cd.