Fixed minor trouble