update with unit test fix