Fixing invalid return type.