diff --git a/src/index.js b/src/index.js
index 4f58ab4..ec53534 100644
--- a/src/index.js
+++ b/src/index.js
@@ -63,7 +63,8 @@ const isInAssignmentContext = path => {
 
     if (
       parentPath.isVariableDeclaration() ||
-      parentPath.isAssignmentExpression()
+      parentPath.isAssignmentExpression() ||
+      parentPath.isClass()
     ) {
       return true
     }
diff --git a/test/fixtures/annotate/class-declaration-dynamic-extends/code.js b/test/fixtures/annotate/class-declaration-dynamic-extends/code.js
new file mode 100644
index 0000000..dea8cd9
--- /dev/null
+++ b/test/fixtures/annotate/class-declaration-dynamic-extends/code.js
@@ -0,0 +1,8 @@
+const base = bar =>
+  class {
+    foo() {
+      return bar
+    }
+  }
+
+class MyClass extends base() {}
diff --git a/test/fixtures/annotate/class-declaration-dynamic-extends/output.js b/test/fixtures/annotate/class-declaration-dynamic-extends/output.js
new file mode 100644
index 0000000..d4529c3
--- /dev/null
+++ b/test/fixtures/annotate/class-declaration-dynamic-extends/output.js
@@ -0,0 +1,6 @@
+const base = bar => class {
+  foo() {
+    return bar;
+  }
+};
+class MyClass extends /*#__PURE__*/base() {}
\ No newline at end of file
diff --git a/test/fixtures/annotate/class-expression-dynamic-extends/code.js b/test/fixtures/annotate/class-expression-dynamic-extends/code.js
new file mode 100644
index 0000000..588cd13
--- /dev/null
+++ b/test/fixtures/annotate/class-expression-dynamic-extends/code.js
@@ -0,0 +1,8 @@
+const base = bar =>
+  class {
+    foo() {
+      return bar
+    }
+  }
+
+const MyClass = class extends base() {}
diff --git a/test/fixtures/annotate/class-expression-dynamic-extends/output.js b/test/fixtures/annotate/class-expression-dynamic-extends/output.js
new file mode 100644
index 0000000..636f15a
--- /dev/null
+++ b/test/fixtures/annotate/class-expression-dynamic-extends/output.js
@@ -0,0 +1,6 @@
+const base = bar => class {
+  foo() {
+    return bar;
+  }
+};
+const MyClass = class extends /*#__PURE__*/base() {};
\ No newline at end of file