lib/daybreak/db.rb in daybreak-0.0.3 vs lib/daybreak/db.rb in daybreak-0.0.4

@@ -133,10 +133,10 @@
 
     # Compact the database to remove stale commits and reduce the file size.
     def compact!
       # Create a new temporary file
       tmp_file = Tempfile.new File.basename(@file_name)
-      copy_db  = DB.new tmp_file.path
+      copy_db  = self.class.new tmp_file.path
 
       # Copy the database key by key into the temporary table
       each do |key, i|
         copy_db.set(key, get(key))
       end