as per my experience baleno head is more powerful then Honda.you can check is your fan is auto? and thermostat valve is installed, after all precautions you can check with mechanic to change pistons rings and clutch plates. you baleno will be like a zero meter again.
if you swap engine you have to swap ECU too and new engine number update on your book. second replaced engine isn't new box packed .new engine is also used one. so don't try this procedure is only recommended for Honda .