Fix test script

Signed-off-by: Amit Gupta <agupta@pivotallabs.com>