Fix allocation fail
Merge pull request #54 from jeiea/python3

Fix allocation fail