update to work with locomotive engine