diff --git a/__init__.py b/__init__.py
index 0a4a066..758e6c0 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .vggish import Vggish
+from .torch_vggish import Vggish
 
 
 def vggish(weights_path: str = None, framework: str = 'pytorch'):