fix argument and UnitTest