diff --git a/CATS/layers/__init__.py b/CATS/layers/__init__.py
index 5af2406..bb67a43 100644
--- a/CATS/layers/__init__.py
+++ b/CATS/layers/__init__.py
@@ -1 +1 @@
-from core import *
+from .core import *
diff --git a/CATS/layers/core.py b/CATS/layers/core.py
index d37b1ac..c95218d 100644
--- a/CATS/layers/core.py
+++ b/CATS/layers/core.py
@@ -2,7 +2,7 @@
 
 import torch
 import torch.nn as nn
-from activation import activation_layer
+from .activation import activation_layer
 
 
 class DNN(nn.Module):