Ensuring the correct return type.