Fix up all test models.

[Closes #2]