From 1a026f69cb62321bdca817ad406e12ae98e04fc6 Mon Sep 17 00:00:00 2001
From: Selimaj Dev <klesti.selimay@icloud.com>
Date: Fri, 22 May 2026 03:44:44 +0200
Subject: [PATCH] Fixed errors

---
 mist-test/src/app.mist | 50 +++++++++++++++++++++---------------------
 src/builder.rs         | 10 ++++++---
 2 files changed, 32 insertions(+), 28 deletions(-)

diff --git a/mist-test/src/app.mist b/mist-test/src/app.mist
index e4c0e5d..9e68eb4 100644
--- a/mist-test/src/app.mist
+++ b/mist-test/src/app.mist
@@ -91,33 +91,33 @@ pub class Application {
             }
         }
     }
+}
 
-    pub i32 loops_test() {
-        for (x : 0..10) {
-            println!("{x}");
-        }
-
-        for (var mut i = 0; i < 10; i++) {
-            println!("{i}");
-        }
-
-        var i = 0;
-
-        while (i == 0) {
-            println!("hi");
-        }
-
-        loop {
-            println!("loop!");
-        }
-
-        if (x == 0) bro();
-        else bro();
-
+i32 test_stmt() {
+    if (0 == 0) {
         0
     }
+    else {
+        1
+    }
 }
 
-pub i32 bro() {
-        0
-    }
\ No newline at end of file
+void loops_test() {
+    for (x : 0..10) {
+        println!("{x}");
+    }
+
+    for (var mut i = 0; i < 10; i++) {
+        println!("{i}");
+    }
+
+    var i = 0;
+
+    while (i == 1) {
+        println!("hi");
+    }
+
+    loop {
+        println!("loop!");
+    }
+}
diff --git a/src/builder.rs b/src/builder.rs
index 34afc8a..f59da2d 100644
--- a/src/builder.rs
+++ b/src/builder.rs
@@ -1,7 +1,7 @@
 use std::{
     collections::HashMap,
     env, fs,
-    path::PathBuf,
+    path::{MAIN_SEPARATOR, PathBuf},
     process::{Command, Stdio},
 };
 
@@ -44,6 +44,8 @@ pub fn build(mut args: Vec<String>, root: PathBuf) -> bool {
 
     let mut mapping = HashMap::new();
 
+    let mist_src = format!(".mist{MAIN_SEPARATOR}src");
+
     for message in cargo_metadata::Message::parse_stream(&mut reader) {
         match message {
             Ok(Message::CompilerMessage(msg)) => {
@@ -53,8 +55,10 @@ pub fn build(mut args: Vec<String>, root: PathBuf) -> bool {
 
                         let mist_file = span
                             .file_name
-                            .replacen(".mist/src", "src", 1)
-                            .replace(".rs", ".mist");
+                            .replacen(&mist_src, "src", 1)
+                            .trim_end_matches(".rs")
+                            .to_string()
+                            + ".mist";
 
                         let mist_path = root.join(&mist_file);