diff --git a/src/main.rs b/src/main.rs
index dbf8eb9..84dae81 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,5 @@
 fn add(a: i32, b: i32) -> i32 {
-    return a + b;
+    a + b
 }
 
 fn main() {
@@ -12,6 +12,6 @@ mod tests {
 
     #[test]
     fn test_add() {
-        assert_eq!(add(2, 3), 5);
+        assert_eq!(add(2, 3), 6);
     }
 }