Fixing deprecated module import