Fixed the trafotron script