1 files changed, 4 insertions, 2 deletions
diff --git a/include/ImageViewer.hpp b/include/ImageViewer.hpp
index f886709..cd1bc8f 100644
--- a/include/ImageViewer.hpp
+++ b/include/ImageViewer.hpp
@@ -46,9 +46,9 @@ namespace QuickMedia {
 
     class ImageViewer {
     public:
-        ImageViewer(MangaImagesPage *manga_images_page, const std::string &content_title, const std::string &chapter_title, int current_page, const Path &chapter_cache_dir);
+        ImageViewer(sf::RenderWindow *window, MangaImagesPage *manga_images_page, const std::string &content_title, const std::string &chapter_title, int current_page, const Path &chapter_cache_dir);
         ~ImageViewer();
-        ImageViewerAction draw(sf::RenderWindow &window);
+        ImageViewerAction draw();
         // Returns page as 1 indexed
         int get_focused_page() const;
         int get_num_pages() const { return num_pages; }
@@ -57,6 +57,8 @@ namespace QuickMedia {
         bool render_page(sf::RenderWindow &window, int page, double offset_y);
         sf::Vector2<double> get_page_size(int page);
     private:
+        sf::RenderWindow *window;
+
         int current_page;
         int num_pages;
         int page_center;