added missing return type annotation