Fix bug in TWFYthon