Small warning fix v. 1.1