Added a missing return call