Fixed minor problem