diff --git a/env.py b/env.py
index 27101ba..df11fc6 100644
--- a/env.py
+++ b/env.py
@@ -4,8 +4,7 @@ import numpy as np
 import pygame
 import random as random_dont_use
 import math
-import entities
-import observables
+from . import entities, observables
 
 
 class ColumbusEnv(gym.Env):