Revert mistakenly added change to py2 init function.

Apologies for the last PR, I mistakenly changed the py2 init function along with the py3 version, the py2 version was correct before.