diff --git a/upload/source/function/function_comment.php b/upload/source/function/function_comment.php
index 2bb1d3ad7e2c9125a7119f373f8da25942fcf6a3..9e9fb0cafb9542e901717524c315d100d6e6b45f 100644
--- a/upload/source/function/function_comment.php
+++ b/upload/source/function/function_comment.php
@@ -70,9 +70,9 @@ function add_comment($message, $id, $idtype, $cid = 0) {
 
 			$album = array();
 			if($pic['albumid']) {
-				$query = C::t('home_album')->fetch($pic['albumid']);
-				if(!$query['albumid']) {
-					C::t('home_pic')->update_for_albumid($albumid, array('albumid' => 0));
+				$album = C::t('home_album')->fetch($pic['albumid']);
+				if(!$album['albumid']) {
+					C::t('home_pic')->update_for_albumid($pic['albumid'], array('albumid' => 0));
 				}
 			}