Small warning fix