fix integration with kms