diff --git a/et_replay/README.md b/et_replay/README.md
index b44cbda7..ce9b4a9a 100644
--- a/et_replay/README.md
+++ b/et_replay/README.md
@@ -5,12 +5,11 @@
 To install `et_replay`, use the following commands:
 
 ```bash
-$ git clone --recurse-submodules git@github.com:facebookresearch/param.git
+$ git clone https://github.com/pytorch-labs/chakra_replay/
 $ conda create -n et_replay python=3.10
 $ conda activate et_replay
-$ cd param
+$ cd chakra_replay
 $ pip3 install -r requirements.txt
-$ cd et_replay
 $ pip3 install .
 ```
 
diff --git a/et_replay/__init__.py b/et_replay/__init__.py
index e9864914..e69de29b 100644
--- a/et_replay/__init__.py
+++ b/et_replay/__init__.py
@@ -1,3 +0,0 @@
-from et_replay.execution_trace import ExecutionTrace
-
-__all__ = ["ExecutionTrace"]
diff --git a/et_replay/pyproject.toml b/et_replay/pyproject.toml
index 24838414..dad804b2 100644
--- a/et_replay/pyproject.toml
+++ b/et_replay/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
 ]
 
 [tool.setuptools.package-dir]
-"et_replay" = "."
+"et_replay" = "et_replay"
 
 [project.scripts]
 comm_replay = "et_replay.tools.comm_replay:main"
diff --git a/et_replay/tests/test_execution_trace.py b/et_replay/tests/test_execution_trace.py
index 43a4589f..fea04304 100644
--- a/et_replay/tests/test_execution_trace.py
+++ b/et_replay/tests/test_execution_trace.py
@@ -3,7 +3,7 @@
 import os
 import unittest
 
-from et_replay import ExecutionTrace
+from et_replay.execution_trace import ExecutionTrace
 from et_replay.tools.validate_trace import TraceValidator
 
 CURR_DIR = os.path.dirname(os.path.realpath(__file__))