diff --git a/CHANGELOG.md b/CHANGELOG.md
index aaba87c..9e23e76 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,12 @@
+## [2.0.1](https://github.com/kwonoj/rx-sandbox/compare/v2.0.0...v2.0.1) (2021-05-18)
+
+
+### Bug Fixes
+
+* **getmessages:** update return type ([c2037bd](https://github.com/kwonoj/rx-sandbox/commit/c2037bd5048facab8d497dabee06221b03d82494))
+
+
+
 # [2.0.0](https://github.com/kwonoj/rx-sandbox/compare/v2.0.0-beta.3...v2.0.0) (2021-05-18)
 
 
diff --git a/README.md b/README.md
index 6dab452..137be30 100644
--- a/README.md
+++ b/README.md
@@ -5,7 +5,7 @@
 # RxSandbox
 
 `RxSandbox` is test suite for RxJS, based on marble diagram DSL for easier assertion around Observables.
-For RxJS 5 support, check pre-1.x versions. 1.x supports latest RxJS 6.x. 2.0.0-beta.* is available to support latest `rxjs@7.x` beta version.
+For RxJS 5 support, check pre-1.x versions. 1.x supports latest RxJS 6.x. 2.* is available to support `rxjs@7.0.1` and above.
 
 ## What's difference with `TestScheduler` in RxJS?
 
diff --git a/package-lock.json b/package-lock.json
index 8cdb995..c0ec2bc 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,6 +1,6 @@
 {
   "name": "rx-sandbox",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {
diff --git a/package.json b/package.json
index 8933d8a..b9676d6 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "rx-sandbox",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "Marble diagram DSL based test suite for RxJS 6",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
diff --git a/src/scheduler/createTestScheduler.ts b/src/scheduler/createTestScheduler.ts
index 764a3a4..f3a02d4 100644
--- a/src/scheduler/createTestScheduler.ts
+++ b/src/scheduler/createTestScheduler.ts
@@ -226,11 +226,14 @@ function getSchedulerFlushFunctions(state: SandboxState, flushWithAsyncTick: boo
   return { flushUntil, advanceTo };
 }
 
-type getMessages = <T = string>(observable: Observable<T>, unsubscriptionMarbles?: string | null) => void;
+type getMessages = <T = string>(
+  observable: Observable<T>,
+  unsubscriptionMarbles?: string | null
+) => Array<TestMessage<T | Array<TestMessage<T>>>>;
 type getMessagesWithTick = <T = string>(
   observable: Observable<T>,
   unsubscriptionMarbles?: string | null
-) => Promise<void>;
+) => Promise<Array<TestMessage<T | Array<TestMessage<T>>>>>;
 
 /**
  * create getMessages function. Depends on flush, this'll either work asynchronously or synchronously.