update unit tests to resolve code for #223 and #227