fixed error: changed return type on wrong function