Change module import to squash a code smell