Added missing return, doh.