diff --git a/pyTsetlinMachineParallel/tm.py b/pyTsetlinMachineParallel/tm.py
index 084db69..018bb64 100644
--- a/pyTsetlinMachineParallel/tm.py
+++ b/pyTsetlinMachineParallel/tm.py
@@ -127,7 +127,7 @@ def __getstate__(self):
 
 	def __setstate__(self, state):
 		self.__dict__.update(state)
-		self.mc_ctm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, self.number_of_patches, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses, self.clause_drop_p, self.literal_drop_p)
+		self.mc_ctm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, self.number_of_patches, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses)
 		self.set_state(state['mc_ctm_state'])
 
 	def __del__(self):
@@ -260,7 +260,7 @@ def __getstate__(self):
 
 	def __setstate__(self, state):
 		self.__dict__.update(state)
-		self.mc_tm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, 1, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses, self.clause_drop_p, self.literal_drop_p)
+		self.mc_tm = _lib.CreateMultiClassTsetlinMachine(self.number_of_classes, self.number_of_clauses, self.number_of_features, 1, self.number_of_ta_chunks, self.number_of_state_bits, self.T, self.s, self.s_range, self.boost_true_positive_feedback, self.weighted_clauses)
 		self.set_state(state['mc_tm_state'])
 
 	def __del__(self):