correct missing return

small correction