300 312 314 316 317

I have JD 300 with the Kohler K341. I think the engine is on its last leg. So I was thinking of swapping it out with a Kohler K321 that I have laying around. Will there be any issues with installing the K321 or will it bolt right up and be an easy swap? Thanks in advance.