Updates for 3.1 compatibility