Fixed coding errors in unit example