diff --git a/pts/__init__.py b/pts/__init__.py
index a6c8795..e69de29 100644
--- a/pts/__init__.py
+++ b/pts/__init__.py
@@ -1,2 +0,0 @@
-from .exception import assert_pts
-from .trainer import Trainer
diff --git a/pts/dataset/__init__.py b/pts/dataset/__init__.py
index 3ba88c4..c55d88b 100644
--- a/pts/dataset/__init__.py
+++ b/pts/dataset/__init__.py
@@ -1,11 +1,11 @@
-from pts.dataset.common import DataEntry, FieldName
-from pts.dataset.list_dataset import ListDataset
-from pts.dataset.loader import DataLoader, InferenceDataLoader, TrainDataLoader
-from pts.dataset.sampler import (
+from .common import DataEntry, FieldName
+from .list_dataset import ListDataset
+from .loader import DataLoader, InferenceDataLoader, TrainDataLoader
+from .sampler import (
+    InstanceSampler,
     BucketInstanceSampler,
     ExpectedNumInstanceSampler,
-    InstanceSampler,
     TestSplitSampler,
     UniformSplitSampler,
 )
-from pts.dataset.utils import to_pandas
+from .utils import to_pandas
diff --git a/pts/feature/__init__.py b/pts/feature/__init__.py
index e4bfaee..778abdb 100644
--- a/pts/feature/__init__.py
+++ b/pts/feature/__init__.py
@@ -1,5 +1,5 @@
-from pts.feature.lag import get_lags_for_frequency
-from pts.feature.time_feature import (
+from .lag import get_lags_for_frequency
+from .time_feature import (
     DayOfMonth,
     DayOfWeek,
     DayOfYear,
@@ -10,7 +10,7 @@ from pts.feature.time_feature import (
     WeekOfYear,
     time_features_from_frequency_str,
 )
-from pts.feature.transform import (
+from .transform import (
     AddAgeFeature,
     AddConstFeature,
     AddObservedValuesIndicator,
diff --git a/pts/feature/transform.py b/pts/feature/transform.py
index a9e9be1..ed152b7 100644
--- a/pts/feature/transform.py
+++ b/pts/feature/transform.py
@@ -6,8 +6,9 @@ from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple
 import numpy as np
 import pandas as pd
 
-from pts import assert_pts
-from pts.dataset import DataEntry, InstanceSampler
+from pts.exception import assert_pts
+from pts.dataset import DataEntry
+from pts.dataset.sampler import InstanceSampler
 
 from .time_feature import TimeFeature
 
diff --git a/pts/model/__init__.py b/pts/model/__init__.py
index 67ebdcf..81a58d9 100644
--- a/pts/model/__init__.py
+++ b/pts/model/__init__.py
@@ -1,4 +1,4 @@
-from pts.model.estimator import Estimator, PTSEstimator
-from pts.model.forecast import Forecast
-from pts.model.predictor import Predictor
-from pts.model.quantile import Quantile
+from .estimator import Estimator, PTSEstimator
+from .forecast import Forecast
+from .predictor import Predictor
+from .quantile import Quantile
diff --git a/pts/trainer.py b/pts/trainer.py
index 426fa63..6a2180f 100644
--- a/pts/trainer.py
+++ b/pts/trainer.py
@@ -5,7 +5,7 @@ import torch
 import torch.nn as nn
 from tqdm import tqdm
 
-from .dataset import TrainDataLoader
+from pts.dataset import TrainDataLoader
 
 
 class Trainer: