made it work with changes to engine