diff --git a/pkg/scheduler/db/models/base_test.go b/pkg/scheduler/db/models/base_test.go
index 4a2deaf646..8c084510be 100644
--- a/pkg/scheduler/db/models/base_test.go
+++ b/pkg/scheduler/db/models/base_test.go
@@ -155,7 +155,8 @@ func TestGuestDisk(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	size, err := gst.(*Guest).DiskSize()
+	onlyLocal := false
+	size, err := gst.(*Guest).DiskSize(onlyLocal)
 	if err != nil {
 		t.Fatal(err)
 	}