Validate key_type as a required parameter for some cases