amend return type for some functions