diff --git a/crates/ide_assists/src/handlers/remove_dbg.rs b/crates/ide_assists/src/handlers/remove_dbg.rs
index 2bf005913a..2862cfa9c2 100644
--- a/crates/ide_assists/src/handlers/remove_dbg.rs
+++ b/crates/ide_assists/src/handlers/remove_dbg.rs
@@ -475,33 +475,29 @@ fn foo() {
         );
         check_assist(
             remove_dbg,
-            r#"fn foo() {
-let test = $0dbg!();
+            r#"
+fn foo() {
+    let test = $0dbg!();
 }"#,
-            r#"fn foo() {
-let test = ();
+            r#"
+fn foo() {
+    let test = ();
 }"#,
         );
         check_assist(
             remove_dbg,
-            r#"fn foo() {
-$0dbg!()
+            r#"
+fn foo() {
+    let t = {
+        println!("Hello, world");
+        $0dbg!()
+    };
 }"#,
-            r#"fn foo() {
-}"#,
-        );
-        check_assist(
-            remove_dbg,
-            r#"fn foo() {
-let t = {
-    println!("Hello, world");
-    $0dbg!()
-};
-}"#,
-            r#"fn foo() {
-let t = {
-    println!("Hello, world");
-};
+            r#"
+fn foo() {
+    let t = {
+        println!("Hello, world");
+    };
 }"#,
         );
     }