Fix Python2 specific name.

Change-Id: I77ac0436b7874b0a738990b8a4a92d901b5550fb