Fix issue in pci-ml model