diff --git a/CalendarNotepad/Extends/DbContext.cs b/CalendarNotepad/Extends/DbContext.cs
index 44fb97e..9cd1b78 100644
--- a/CalendarNotepad/Extends/DbContext.cs
+++ b/CalendarNotepad/Extends/DbContext.cs
@@ -12,7 +12,7 @@ namespace CalendarNotepad.Extends
 
         }
 
-        public static ConnectionConfig GetConfig() => new ConnectionConfig {
+        public static ConnectionConfig GetConfig() => new() {
             DbType = DbType.Sqlite,
             ConnectionString = AppConfig.SqliteFileName,
             IsAutoCloseConnection = false,
diff --git a/CalendarNotepad/Models/WorkUnit.cs b/CalendarNotepad/Models/WorkUnit.cs
index 913097c..b1f590a 100644
--- a/CalendarNotepad/Models/WorkUnit.cs
+++ b/CalendarNotepad/Models/WorkUnit.cs
@@ -1,8 +1,11 @@
-namespace CalendarNotepad.Models
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace CalendarNotepad.Models
 {
     /// <summary>
     /// 工作记录单元
     /// </summary>
+    [Table("WorkUnit")]
     public class WorkUnit
     {
         /// <summary>