diff --git a/kiwi_generator/CHANGELOG.md b/kiwi_generator/CHANGELOG.md
index 6aeefed..b505484 100644
--- a/kiwi_generator/CHANGELOG.md
+++ b/kiwi_generator/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.3.1
+### Fixed
+* Upgrade dependencies and fix a deprecated error.
+
 ## 0.3.0
 ### Fixed
 * Upgrade dependencies
diff --git a/kiwi_generator/lib/src/injector_generator.dart b/kiwi_generator/lib/src/injector_generator.dart
index 3f320fd..34711fc 100644
--- a/kiwi_generator/lib/src/injector_generator.dart
+++ b/kiwi_generator/lib/src/injector_generator.dart
@@ -24,7 +24,7 @@ class InjectorGenerator extends Generator {
   String generate(LibraryReader library, BuildStep buildStep) {
     // An injector is an abstract class where all abstract methods are
     // annotated with Register.
-    final injectors = library.classElements
+    final injectors = library.classes
         .where((c) =>
             c.isAbstract &&
             c.methods.where((m) => m.isAbstract).isNotEmpty &&
diff --git a/kiwi_generator/pubspec.yaml b/kiwi_generator/pubspec.yaml
index 468cb5d..d5231bc 100644
--- a/kiwi_generator/pubspec.yaml
+++ b/kiwi_generator/pubspec.yaml
@@ -1,6 +1,6 @@
 name: kiwi_generator
 description: Generates dependency injection code using the kiwi package to reduce development time.
-version: 0.3.0
+version: 0.3.1
 author: Romain Rastel <lets4r@gmail.com>
 homepage: https://github.com/letsar/kiwi
 
@@ -15,7 +15,7 @@ dependencies:
   dart_style: '>= 1.0.0 <1.3.0'
   kiwi: '>=0.1.0 <0.2.0'
   path: ^1.6.2
-  source_gen: ^0.9.1
+  source_gen: ^0.9.4
 
 dev_dependencies:
   build_runner: ^1.1.0