diff --git a/themes/danbooru/view.theme.php b/themes/danbooru/view.theme.php
index 7491d7d3..1a020b7b 100644
--- a/themes/danbooru/view.theme.php
+++ b/themes/danbooru/view.theme.php
@@ -1,6 +1,6 @@
 <?php
 
-class CustomViewTheme extends ViewTheme {
+class CustomViewImageTheme extends ViewImageTheme {
 	public function display_image_not_found($page, $image_id) {
 		$page->set_title("Image not found");
 		$page->set_heading("Image not found");