Fix type in MPS version