Update import for 3.0