updated test to reflect return type for python 3