diff --git a/Source/classes/RelatedChangesetsColumn.class.php b/Source/classes/RelatedChangesetsColumn.class.php
index f3c0a83..a23683c 100644
--- a/Source/classes/RelatedChangesetsColumn.class.php
+++ b/Source/classes/RelatedChangesetsColumn.class.php
@@ -18,7 +18,7 @@
 		plugin_pop_current();
 	}
 
-	public function cache( $p_bugs ) {
+	public function cache( array $p_bugs ) {
 		if ( count( $p_bugs ) < 1 ) {
 			return;
 		}
@@ -45,7 +45,7 @@
 		}
 	}
 
-	public function display( $p_bug, $p_columns_target ) {
+	public function display( BugData $p_bug, $p_columns_target ) {
 		plugin_push_current( 'Source' );
 
 		if ( isset( $this->changeset_cache[ $p_bug->id ] ) ) {