minor edit: fixing fmt and test