improving unit test
minor change