small junit improvement