added missing return stmt