Fixing methods to match train controller