1 failing test - need to get passing.
Fixed failing unit test