A small change with unit testing.