Timeline server domain not set correctly when using shell_command on Windows

Running org.apache.hadoop.yarn.applications.distributedshell.TestDistributedShell
Tests run: 1, Failures: 1, Errors: 0, Skipped: 0, Time elapsed: 37.661 sec <<< FAILURE! - in org.apache.hadoop.yarn.applications.distribut
testDSShellWithDomain(org.apache.hadoop.yarn.applications.distributedshell.TestDistributedShell)  Time elapsed: 37.366 sec  <<< FAILURE!
org.junit.ComparisonFailure: expected:<[TEST_DOMAIN]> but was:<[DEFAULT]>
        at org.junit.Assert.assertEquals(Assert.java:115)
        at org.junit.Assert.assertEquals(Assert.java:144)
        at org.apache.hadoop.yarn.applications.distributedshell.TestDistributedShell.testDSShell(TestDistributedShell.java:290)
        at org.apache.hadoop.yarn.applications.distributedshell.TestDistributedShell.testDSShellWithDomain(TestDistributedShell.java:179)