Fixed mismatch implementation return type