Fix tdk.py