Added missing return type to fix CI tests.