Added include to fix compile error on Ubuntu.