1 files changed, 4 insertions, 3 deletions
diff --git a/Source/SPSSHTunnel.h b/Source/SPSSHTunnel.h
index bf7fd8a4..231a41a6 100644
--- a/Source/SPSSHTunnel.h
+++ b/Source/SPSSHTunnel.h
@@ -23,9 +23,10 @@
 //
 //  More info at <http://code.google.com/p/sequel-pro/>
 
-#import <MCPKit/MCPKit.h>
+#import "SPMySQLConnectionProxy.h"
+#import "SPMySQLConstants.h"
 
-@interface SPSSHTunnel : NSObject <MCPConnectionProxy>
+@interface SPSSHTunnel : NSObject <SPMySQLConnectionProxy>
 {
 	id delegate;
 
@@ -78,7 +79,7 @@
 - (BOOL)setPasswordKeychainName:(NSString *)theName account:(NSString *)theAccount;
 - (BOOL)setPassword:(NSString *)thePassword;
 - (BOOL)setKeyFilePath:(NSString *)thePath;
-- (NSInteger)state;
+- (SPMySQLConnectionProxyState)state;
 - (NSString *)lastError;
 - (NSString *)debugMessages;
 - (NSUInteger)localPort;