When to go through immigration for transit visa?

I have a booking from Singapore to Kuala Lumpur on to Shanghai, coming back direct to Singapore on separate tickets but with the same airline for the first two flights. I would like to use the 72 hour free transit. When I arrive in Kuala Lumpur, should I go through immigration to officially enter Malaysia or can I just transit directly to the Shanghai flight?
Thanks!