adjust policy regarding to threads

1 files changed, 3 insertions, 3 deletions

diff --git a/aiogreen.py b/aiogreen.py
index 2d5ec6f..a9e489f 100644
--- a/aiogreen.py
+++ b/aiogreen.py
@@ -40,7 +40,7 @@ class EventLoopPolicy(asyncio.AbstractEventLoopPolicy):
 
     def get_event_loop(self):
         if not _is_main_thread():
-            raise NotImplementedError("currently aiogreen can only run in the main thread")
+            return None
         if self._loop is None:
             self._loop = EventLoop()
         return self._loop
@@ -49,8 +49,8 @@ class EventLoopPolicy(asyncio.AbstractEventLoopPolicy):
         return EventLoop()
 
     def set_event_loop(self, loop):
-        if self._loop is not None:
-            raise NotImplementedError("cannot run two event loops in the same thread")
+        if not _is_main_thread():
+            raise NotImplementedError("aiogreen can only run in the main thread")
         self._loop = loop