refining unit tests