Add tests for target units