fixing import to eliminate a deprecation warning