diff --git a/nandroid.h b/nandroid.h
index 3f6c5a8..2ab0308 100644
--- a/nandroid.h
+++ b/nandroid.h
@@ -3,6 +3,6 @@
 
 int nandroid_main(int argc, char** argv);
 int nandroid_backup(char* backup_path);
-int nandroid_restore(char* backup_path);
+int nandroid_restore(char* backup_path, int restore_boot, int restore_system, int restore_data, int restore_cache);
 
 #endif
\ No newline at end of file