Update test for improved engine behaviour.
Merge pull request #80 from rhattersley/update-test

Update test for improved engine behaviour.