struct print_entry *entry;
        unsigned long irq_flags;
        struct page *pages[2];
+       void *map_page[2];
        int nr_pages = 1;
        ssize_t written;
-       void *page1;
-       void *page2;
        int offset;
        int size;
        int len;
        int ret;
+       int i;
 
        if (tracing_disabled)
                return -EINVAL;
                goto out;
        }
 
-       page1 = kmap_atomic(pages[0]);
-       if (nr_pages == 2)
-               page2 = kmap_atomic(pages[1]);
+       for (i = 0; i < nr_pages; i++)
+               map_page[i] = kmap_atomic(pages[i]);
 
        local_save_flags(irq_flags);
        size = sizeof(*entry) + cnt + 2; /* possible \n added */
 
        if (nr_pages == 2) {
                len = PAGE_SIZE - offset;
-               memcpy(&entry->buf, page1 + offset, len);
-               memcpy(&entry->buf[len], page2, cnt - len);
+               memcpy(&entry->buf, map_page[0] + offset, len);
+               memcpy(&entry->buf[len], map_page[1], cnt - len);
        } else
-               memcpy(&entry->buf, page1 + offset, cnt);
+               memcpy(&entry->buf, map_page[0] + offset, cnt);
 
        if (entry->buf[cnt - 1] != '\n') {
                entry->buf[cnt] = '\n';
        *fpos += written;
 
  out_unlock:
-       if (nr_pages == 2)
-               kunmap_atomic(page2);
-       kunmap_atomic(page1);
-       while (nr_pages > 0)
-               put_page(pages[--nr_pages]);
+       for (i = 0; i < nr_pages; i++){
+               kunmap_atomic(map_page[i]);
+               put_page(pages[i]);
+       }
  out:
        return written;
 }