lib/time_clock/shift.rb in time_clock-0.0.3 vs lib/time_clock/shift.rb in time_clock-0.0.4

@@ -5,10 +5,10 @@
 
     attr_reader :start_time, :end_time
 
     def initialize(start_time, end_time)
       raise EndTimeAfterStartTimeError unless end_time > start_time
-      @start_time, @end_time = start_time, end_time
+      @start_time, @end_time = start_time.to_time, end_time.to_time
     end
 
     def overlaps?(shift)
       start_time <= shift.end_time && end_time >= shift.start_time
     end
\ No newline at end of file