diff --git a/include/transport/OAuth2.h b/include/transport/OAuth2.h
index 92c7c7a832f26bcf0a9c1fb3358b4e812508e904..c8612d17c1caa3cb8789c2a01f20e35d9ed667ca 100644
--- a/include/transport/OAuth2.h
+++ b/include/transport/OAuth2.h
@@ -35,7 +35,11 @@ class OAuth2 {
 
 		std::string generateAuthURL();
 
-		std::string handleOAuth2Code(const std::string &code, const std::string &state);
+		const std::string &getState() {
+			return m_state;
+		}
+
+		std::string requestToken(const std::string &code, std::string &error);
 
 	private:
 		std::string m_clientId;