final update to unit test