diff --git a/scripts/attention.py b/scripts/attention.py
index 057a04a..f422ae8 100644
--- a/scripts/attention.py
+++ b/scripts/attention.py
@@ -8,11 +8,11 @@
 import torch
 from torch.autograd import Variable
 
-from TransVAE.transvae.trans_models import TransVAE
-from TransVAE.transvae.rnn_models import RNN, RNNAttn
+from transvae.trans_models import TransVAE
+from transvae.rnn_models import RNN, RNNAttn
 
-from TransVAE.transvae.data import vae_data_gen, make_std_mask
-from TransVAE.scripts.parsers import attn_parser
+from transvae.data import vae_data_gen, make_std_mask
+from scripts.parsers import attn_parser
 
 def calc_attention(args):
     ### Load model
diff --git a/scripts/sample.py b/scripts/sample.py
index 7d0dfc8..0659e41 100644
--- a/scripts/sample.py
+++ b/scripts/sample.py
@@ -5,10 +5,10 @@
 import numpy as np
 import pandas as pd
 
-from TransVAE.transvae.trans_models import TransVAE
-from TransVAE.transvae.rnn_models import RNN, RNNAttn
-from TransVAE.transvae.tvae_util import calc_entropy
-from TransVAE.scripts.parsers import sample_parser
+from transvae.trans_models import TransVAE
+from transvae.rnn_models import RNN, RNNAttn
+from transvae.tvae_util import calc_entropy
+from scripts.parsers import sample_parser
 
 def sample(args):
     ### Load model
diff --git a/transvae/version.py b/transvae/version.py
index 16aa1c8..e2b8033 100644
--- a/transvae/version.py
+++ b/transvae/version.py
@@ -4,7 +4,7 @@
 # Format expected by setup.py and doc/source/conf.py: string of form "X.Y.Z"
 _version_major = 0
 _version_minor = 4
-_version_micro = 2  # use '' for first of series, number for 1 and above
+_version_micro = 3  # use '' for first of series, number for 1 and above
 # _version_extra = 'dev1'
 _version_extra = ''  # Uncomment this for full releases