diff --git a/yii/image/drivers/Kohana/Image.php b/yii/image/drivers/Kohana/Image.php
index f701661..41f6b1e 100644
--- a/yii/image/drivers/Kohana/Image.php
+++ b/yii/image/drivers/Kohana/Image.php
@@ -500,14 +500,14 @@ public function reflection($height = NULL, $opacity = 100, $fade_in = FALSE)
          *     $mark = Image::factory('upload/watermark.png');
          *     $image->watermark($mark, TRUE, TRUE);
          *
-         * @param   Image    $watermark  watermark Image instance
-         * @param   integer  $offset_x   offset from the left
-         * @param   integer  $offset_y   offset from the top
-         * @param   integer  $opacity    opacity of watermark: 1-100
+         * @param   Kohana_Image  $watermark  watermark Image instance
+         * @param   integer       $offset_x   offset from the left
+         * @param   integer       $offset_y   offset from the top
+         * @param   integer       $opacity    opacity of watermark: 1-100
          * @return  $this
          * @uses    Image::_do_watermark
          */
-        public function watermark(Image $watermark, $offset_x = NULL, $offset_y = NULL, $opacity = 100)
+        public function watermark(Kohana_Image $watermark, $offset_x = NULL, $offset_y = NULL, $opacity = 100)
         {
                 if ($offset_x === NULL)
                 {
@@ -724,13 +724,13 @@ abstract protected function _do_reflection($height, $opacity, $fade_in);
         /**
          * Execute a watermarking.
          *
-         * @param   Image    $image     watermarking Image
-         * @param   integer  $offset_x  offset from the left
-         * @param   integer  $offset_y  offset from the top
-         * @param   integer  $opacity   opacity of watermark
+         * @param   Kohana_Image  $image     watermarking Kohana_Image
+         * @param   integer       $offset_x  offset from the left
+         * @param   integer       $offset_y  offset from the top
+         * @param   integer       $opacity   opacity of watermark
          * @return  void
          */
-        abstract protected function _do_watermark(Image $image, $offset_x, $offset_y, $opacity);
+        abstract protected function _do_watermark(Kohana_Image $image, $offset_x, $offset_y, $opacity);
 
         /**
          * Execute a background.
@@ -761,4 +761,4 @@ abstract protected function _do_save($file, $quality);
          */
         abstract protected function _do_render($type, $quality);
 
-} // End Image
\ No newline at end of file
+} // End Image
diff --git a/yii/image/drivers/Kohana/Image/GD.php b/yii/image/drivers/Kohana/Image/GD.php
index fd19798..697ca83 100644
--- a/yii/image/drivers/Kohana/Image/GD.php
+++ b/yii/image/drivers/Kohana/Image/GD.php
@@ -443,13 +443,13 @@ protected function _do_reflection($height, $opacity, $fade_in)
         /**
          * Execute a watermarking.
          *
-         * @param   Image    $image     watermarking Image
-         * @param   integer  $offset_x  offset from the left
-         * @param   integer  $offset_y  offset from the top
-         * @param   integer  $opacity   opacity of watermark
+         * @param   Kohana_Image  $image     watermarking Kohana_Image
+         * @param   integer       $offset_x  offset from the left
+         * @param   integer       $offset_y  offset from the top
+         * @param   integer       $opacity   opacity of watermark
          * @return  void
          */
-        protected function _do_watermark(Image $watermark, $offset_x, $offset_y, $opacity)
+        protected function _do_watermark(Kohana_Image $watermark, $offset_x, $offset_y, $opacity)
         {
                 if (empty(Image_GD::$_available_functions[Image_GD::IMAGELAYEREFFECT]))
                 {
@@ -665,4 +665,4 @@ protected function _create($width, $height)
                 return $image;
         }
 
-} // End Image_GD
\ No newline at end of file
+} // End Image_GD
diff --git a/yii/image/drivers/Kohana/Image/Imagick.php b/yii/image/drivers/Kohana/Image/Imagick.php
index 36a139a..d0ea610 100644
--- a/yii/image/drivers/Kohana/Image/Imagick.php
+++ b/yii/image/drivers/Kohana/Image/Imagick.php
@@ -202,7 +202,7 @@ protected function _do_reflection($height, $opacity, $fade_in)
                 return FALSE;
         }
 
-        protected function _do_watermark(Image $image, $offset_x, $offset_y, $opacity)
+        protected function _do_watermark(Kohana_Image $image, $offset_x, $offset_y, $opacity)
         {
                 // Convert the Image intance into an Imagick instance
                 $watermark = new Imagick;
@@ -335,4 +335,4 @@ protected function _get_imagetype($extension)
 
                 return array($format, $type);
         }
-} // End Kohana_Image_Imagick
\ No newline at end of file
+} // End Kohana_Image_Imagick