Unit Tests

- Fixed mistake introduced by template upgrade.