Change script to pass in unit test target