diff --git a/README.md b/README.md
index f9267e3..59401e1 100644
--- a/README.md
+++ b/README.md
@@ -9,7 +9,6 @@ Learning with a Stochastic Actor](https://arxiv.org/pdf/1801.01290.pdf) -> [SAC_
 ### Requirements
 ------------
 *   [mujoco-py](https://github.com/openai/mujoco-py)
-*   [TensorboardX](https://github.com/lanpa/tensorboardX)
 *   [PyTorch](http://pytorch.org/)
 
 ### Default Arguments and Usage
diff --git a/main.py b/main.py
index a3047c5..e6ac07b 100644
--- a/main.py
+++ b/main.py
@@ -5,7 +5,7 @@ import numpy as np
 import itertools
 import torch
 from sac import SAC
-from tensorboardX import SummaryWriter
+from torch.utils.tensorboard import SummaryWriter
 from replay_memory import ReplayMemory
 
 parser = argparse.ArgumentParser(description='PyTorch Soft Actor-Critic Args')