Rewrite UnitTest to be more understandable