fixed unit test + added one