1 files changed, 4 insertions, 2 deletions

diff --git a/vm/page.c b/vm/page.c
index b40382f..b98c553 100644
--- a/vm/page.c
+++ b/vm/page.c
@@ -1,10 +1,11 @@
 #include <string.h>
 #include "filesys/file.h"
-#include "userprog/process.h"
 #include "threads/thread.h"
 #include "threads/vaddr.h"
 #include "threads/malloc.h"
 #include "threads/palloc.h"
+#include "userprog/pagedir.h"
+#include "userprog/process.h"
 #include "vm/page.h"
 
 static unsigned page_table_hash (const struct hash_elem *e, void *aux UNUSED);
@@ -52,6 +53,7 @@ static void
 page_table_entry_free (struct hash_elem *e, void *aux UNUSED)
 {
   struct page_table_entry *pte = hash_entry (e, struct page_table_entry, elem);
+  pagedir_clear_page (thread_current ()->pagedir, pte->upage);
   free (pte);
 }
 
@@ -97,7 +99,7 @@ page_table_fetch (struct hash *ht, void *upage)
 }
 
 /* remove an entry from the page table. returns the element or NULL if not found.
-   the caller is responsible for freeing the element! */
+   the caller is responsible for freeing the element and the associated page! */
 struct page_table_entry *
 page_table_remove (struct hash *ht, void *upage)
 {