Updated test to print out units