Fixed potential import error