Re-add test. Current tests are targetted at 1% code.