Added missing return call