diff --git a/mlpp/log_reg/log_reg.cpp b/mlpp/log_reg/log_reg.cpp
index d99a638..2761e33 100644
--- a/mlpp/log_reg/log_reg.cpp
+++ b/mlpp/log_reg/log_reg.cpp
@@ -285,7 +285,7 @@ void MLPPLogReg::initialize() {
 	_initialized = true;
 }
 
-MLPPLogReg::MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg, real_t p_lambda, real_t p_alpha) {
+MLPPLogReg::MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPVector> &p_output_set, MLPPReg::RegularizationType p_reg, real_t p_lambda, real_t p_alpha) {
 	_input_set = p_input_set;
 	_output_set = p_output_set;
 	_n = p_input_set->size().y;
diff --git a/mlpp/log_reg/log_reg.h b/mlpp/log_reg/log_reg.h
index c96385e..849ef78 100644
--- a/mlpp/log_reg/log_reg.h
+++ b/mlpp/log_reg/log_reg.h
@@ -17,9 +17,6 @@
 
 #include "../regularization/reg.h"
 
-#include <string>
-#include <vector>
-
 class MLPPLogReg : public Reference {
 	GDCLASS(MLPPLogReg, Reference);
 
@@ -56,9 +53,7 @@ public:
 	bool is_initialized();
 	void initialize();
 
-	//MLPPSoftmaxReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
-
-	MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
+	MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPVector> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
 
 	MLPPLogReg();
 	~MLPPLogReg();