fix train button to work on iphone