Use the right return type