Fixed improper return for python type