diff --git a/source/main.cpp b/source/main.cpp
index f0631aa..098dfd0 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -16,8 +16,8 @@ class RbmListener : public Rbm::IListener
 	public:
 	RbmListener() {}
 	virtual ~RbmListener() {}
-		
-	bool onProgress(const Rbm::Status &status)
+	
+	bool onProgress(Rbm *pRbm, const Rbm::Status &status)
 	{
 		std::cout << "Progress :          " << status.progress << " %" << std::endl;
 		std::cout << "error (per mini batch) = " << status.err << std::endl;