Revert "changing test function"

This reverts commit 057a1922b662348381bd5c349bfe401af4f959b7.