Add missing return statement from module