Fix Python v3 compatibility