diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 55965c3..8783200 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -84,6 +84,7 @@ jobs:
         ruby-version: "${{ matrix.ruby }}"
     - name: Setup database
       run: |
+        sudo apt-get update
         sudo apt-get install -y postgresql-client
         PGPASSWORD=postgres psql -c 'create database assignable_values_test;' -U postgres -p 5432 -h localhost
     - name: Bundle
diff --git a/CHANGELOG.md b/CHANGELOG.md
index b60740a..141a944 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -12,6 +12,10 @@ This project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.html
 
 -
 
+## 1.0.1 - 2024-08-27
+
+- Hook into railties correctly
+
 ## 1.0.0 - 2024-02-14
 
 - This gem has been in productive use for 12 years now, time for a 1.0 release! 🥳
diff --git a/Gemfile.5.0.lock b/Gemfile.5.0.lock
index fe4fac1..e512d10 100644
--- a/Gemfile.5.0.lock
+++ b/Gemfile.5.0.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    assignable_values (1.0.0)
+    assignable_values (1.0.1)
       activerecord (>= 2.3)
 
 GEM
diff --git a/Gemfile.5.1.lock b/Gemfile.5.1.lock
index 3c555bc..c87ce9c 100644
--- a/Gemfile.5.1.lock
+++ b/Gemfile.5.1.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    assignable_values (1.0.0)
+    assignable_values (1.0.1)
       activerecord (>= 2.3)
 
 GEM
diff --git a/Gemfile.5.1.pg.lock b/Gemfile.5.1.pg.lock
index f367bd1..68457ad 100644
--- a/Gemfile.5.1.pg.lock
+++ b/Gemfile.5.1.pg.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    assignable_values (1.0.0)
+    assignable_values (1.0.1)
       activerecord (>= 2.3)
 
 GEM
diff --git a/Gemfile.6.1.pg.lock b/Gemfile.6.1.pg.lock
index 102b7d4..f34b53e 100644
--- a/Gemfile.6.1.pg.lock
+++ b/Gemfile.6.1.pg.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    assignable_values (1.0.0)
+    assignable_values (1.0.1)
       activerecord (>= 2.3)
 
 GEM
diff --git a/Gemfile.7.1.pg.lock b/Gemfile.7.1.pg.lock
index a991c2e..1ce9d7c 100644
--- a/Gemfile.7.1.pg.lock
+++ b/Gemfile.7.1.pg.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    assignable_values (1.0.0)
+    assignable_values (1.0.1)
       activerecord (>= 2.3)
 
 GEM
diff --git a/lib/assignable_values/active_record.rb b/lib/assignable_values/active_record.rb
index a324327..b2ede63 100644
--- a/lib/assignable_values/active_record.rb
+++ b/lib/assignable_values/active_record.rb
@@ -16,5 +16,6 @@ def belongs_to_association?(property)
   end
 end
 
-ActiveRecord::Base.send(:extend, AssignableValues::ActiveRecord)
-
+ActiveSupport.on_load(:active_record) do
+  extend(AssignableValues::ActiveRecord)
+end
diff --git a/lib/assignable_values/version.rb b/lib/assignable_values/version.rb
index baf3eb8..f8541a7 100644
--- a/lib/assignable_values/version.rb
+++ b/lib/assignable_values/version.rb
@@ -1,3 +1,3 @@
 module AssignableValues
-  VERSION = '1.0.0'
+  VERSION = '1.0.1'
 end