Added a missing return statement