fix code for pycharme