fixes another broken unit test + improves comments