diff --git a/setup.py b/setup.py
index 02dc4e8..e5a208d 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="python-test-bed-adapter",
-    version="0.0.5",
+    version="0.0.7",
     author="Hugo J. Bello",
     author_email="hjbello.wk@gmail.com",
     description="This is the test-bed adapter for Python: it allows you to easily connect Python services to the Apache Kafka test-bed via Python.",
diff --git a/test_bed_adapter/services/__init__.py b/test_bed_adapter/services/__init__.py
new file mode 100644
index 0000000..b702bec
--- /dev/null
+++ b/test_bed_adapter/services/__init__.py
@@ -0,0 +1 @@
+from . import time_service
\ No newline at end of file
diff --git a/test_bed_adapter/services/time_service.py b/test_bed_adapter/services/time_service.py
index e16c61a..ffbcd2c 100644
--- a/test_bed_adapter/services/time_service.py
+++ b/test_bed_adapter/services/time_service.py
@@ -71,7 +71,7 @@ def get_trial_date(self):
         """ Returns UTC date of trial time """
         # Return elapsed if not idle otherwise return date
         elapsed_ms = millisecond_since_date(self.localUpdatedSimTimeAt) * self.trialTimeSpeed
-        return datetime.fromtimestamp((self.trialTime + elapsed_ms) / 1000.0) if self.state is not "Idle" else datetime.now()
+        return datetime.fromtimestamp((self.trialTime + elapsed_ms) / 1000.0) if self.state != "Idle" else datetime.now()
 
     def get_trial_elapsed_time(self):
         """ Returns number of milliseconds elapsed since Unix Epoch """