fixed minor problem