diff --git a/TODO.md b/TODO.md
index 36f2a2c..7e1c98e 100644
--- a/TODO.md
+++ b/TODO.md
@@ -5,3 +5,10 @@
 * SMTP configuration
 
 * UNA's config should be in a gem, but for now we're doing it here so we can test it
+
+* Fix workBase in schema plugin to reference this include back
+```
+#TODO: Work Behavior - Uncomment this
+      #include HykuAddons::TaskMaster::WorkBehavior
+```
+
diff --git a/app/models/una_open_educational_resource.rb b/app/models/una_open_educational_resource.rb
index 0ec7d1a..cb22baf 100644
--- a/app/models/una_open_educational_resource.rb
+++ b/app/models/una_open_educational_resource.rb
@@ -3,7 +3,7 @@ class UnaOpenEducationalResource < ActiveFedora::Base
   include Hyrax::WorkBehavior
   include Hyrax::DOI::DataCiteDOIBehavior
 
-  include HykuAddons::Schema::WorkBase
+  include Hyku::Schema::WorkBase
   include Hyrax::Schema(:una_open_educational_resource)
 
   # Included after other field definitions