Do I need a 904 trans that is specifically built for my 1961 170 ci slant six?

thanks for the help. My current trans shifts a little hard from first to second. I know its a later version of the trans. Its also leaking so I'm thinking about my options. Thanks very much.