diff --git a/.gitignore b/.gitignore
index 795c7fd..7338e99 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,3 @@
 /target
-/test
+/test*
 .rgit
diff --git a/src/base.rs b/src/base.rs
index b23b9f5..8148522 100644
--- a/src/base.rs
+++ b/src/base.rs
@@ -215,6 +215,19 @@ pub fn create_branch(name: String, oid: String) {
     );
 }
 
+pub fn init() -> std::io::Result<()> {
+    data::init()?;
+    data::update_ref(
+        "HEAD".to_owned(),
+        data::RefValue {
+            symbolic: true,
+            value: "refs/heads/master".to_owned(),
+        },
+        true,
+    );
+    Ok(())
+}
+
 fn is_ignored(path: &String) -> bool {
     if path.contains(".rgit") {
         true
diff --git a/src/main.rs b/src/main.rs
index 5c00368..f13f35e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -83,7 +83,7 @@ fn main() {
 }
 
 fn init() {
-    match data::init() {
+    match base::init() {
         Ok(()) => println!("Repository created"),
         _ => println!("Failed. Perhaps the repository already exists."),
     }