4d56 water pump

Hello everyone, I recently got my 4d55 block back from the machine shop, I had them press new liners in, do the finish hone, check and linebore the block, and recondition the connecting rods. The pistons went to a local shop and got ceramic coating on the crowns and skirts to help with the heat, and I purchased a recessed valve late model 4d56t cylinder head to lower my compression (machine shop says it brought me down to 19:1 which is what I’m looking for). Now my question here is which 4d56t water pump should I be purchasing, they seem to be 2 and 4 bolt in design. I have a tc04-10t turbo and I believe the water pump plays a role in the water cooled portion of the pump. Any help would be appreciated!