fixing minor problem