Small warning cleanup