Looks fine though I agree with a PF fold usually. Villain never has a set here, because he'd be terrified about letting a diamond out. He'd raise somewhere. So your hand is going to win here almost 100%. I shove river probably 90%, and check/shove 10%.