fix + feat (#643)