adds the return back in (forgot in prev commit)