Add tests for this.