fix the bug python (my first one)(ch)