Refine previous commit

2 files changed, 4 insertions, 6 deletions

diff --git a/todolist/app.go b/todolist/app.go
index e3d8cc7..1ab6b20 100644
--- a/todolist/app.go
+++ b/todolist/app.go
@@ -15,11 +15,6 @@ type App struct {
 
 func NewApp() *App {
 	app := &App{TodoList: &TodoList{}, TodoStore: NewFileStore()}
-	if app.TodoStore.FileLocation == "" {
-		fmt.Println("No todo file found!")
-		fmt.Println("You may run 'todo init' to initialize an empty repo in current directory.")
-		return nil
-	}
 	return app
 }
 
diff --git a/todolist/file_store.go b/todolist/file_store.go
index 63738b2..4a7f0e0 100644
--- a/todolist/file_store.go
+++ b/todolist/file_store.go
@@ -25,7 +25,10 @@ func NewFileStore() *FileStore {
 	} else if err2 == nil {
 		return &FileStore{FileLocation: homerepo, Loaded: false}
 	} else {
-		return &FileStore{FileLocation: "", Loaded: false}
+		fmt.Println("No todo file found!")
+		fmt.Println("You may run 'todo init' to initialize an empty repo in working directory.")
+		os.Exit(1)
+		return nil
 	}
 }