diff --git a/PBGitRevList.h b/PBGitRevList.h
index 2f71c3f..42bc82a 100644
--- a/PBGitRevList.h
+++ b/PBGitRevList.h
@@ -8,14 +8,15 @@
 
 #import <Cocoa/Cocoa.h>
 
+@class PBGitRepository;
 
 @interface PBGitRevList : NSObject {
 	NSArray* commits;
-	id repository;
+	PBGitRepository *repository;
 	NSString* lastSha;
 }
 
-- initWithRepository:(id)repo;
+- initWithRepository:(PBGitRepository *)repo;
 - (void) readCommitsForce: (BOOL) force;
 - (void) reload;
 
diff --git a/PBGitRevList.mm b/PBGitRevList.mm
index 1d8023a..b2d8a49 100644
--- a/PBGitRevList.mm
+++ b/PBGitRevList.mm
@@ -23,7 +23,7 @@ using namespace std;
 @implementation PBGitRevList
 
 @synthesize commits;
-- initWithRepository: (id) repo
+- (id)initWithRepository:(PBGitRepository *)repo
 {
 	repository = repo;
 	[repository addObserver:self forKeyPath:@"currentBranch" options:0 context:nil];