diff --git a/spec/services/update_timeslices_course_wiki_spec.rb b/spec/services/update_timeslices_course_wiki_spec.rb
index c015128a3f..eee349048f 100644
--- a/spec/services/update_timeslices_course_wiki_spec.rb
+++ b/spec/services/update_timeslices_course_wiki_spec.rb
@@ -98,7 +98,7 @@
 
     it 'updates current and future timeslices if new timeslice duration is smaller' do
       # Update timeslice duration to 12 hours
-      course.flags = { timeslice_duration: 43200 }
+      course.flags = { timeslice_duration: { default: 43200 } }
       course.save
 
       described_class.new(course).run
@@ -112,7 +112,7 @@
 
     it 'updates current and future timeslices if new timeslice duration is greater' do
       # Update timeslice duration to 2 days
-      course.flags = { timeslice_duration: 172800 }
+      course.flags = { timeslice_duration: { default: 172800 } }
       course.save
 
       described_class.new(course).run