Fix --py option