I value shove the flop a lot in these spots.. a ton of his range calls you here.. overpairs and any kind of big FD will call here a lot.

You don't want to raise small like you did and chance peeling a scare card for both of you. Getting the action killed while holding a set sucks.