fix code with python

DO NOT MERGE, this is test code