diff --git a/man/mlr_callback_set.unfreeze.Rd b/man/mlr_callback_set.unfreeze.Rd
index 46dfaf30..8551d640 100644
--- a/man/mlr_callback_set.unfreeze.Rd
+++ b/man/mlr_callback_set.unfreeze.Rd
@@ -34,7 +34,6 @@ Other Callback:
 \itemize{
 \item \href{#method-CallbackSetUnfreeze-new}{\code{CallbackSetUnfreeze$new()}}
 \item \href{#method-CallbackSetUnfreeze-on_begin}{\code{CallbackSetUnfreeze$on_begin()}}
-\item \href{#method-CallbackSetUnfreeze-on_batch_end}{\code{CallbackSetUnfreeze$on_batch_end()}}
 \item \href{#method-CallbackSetUnfreeze-on_epoch_begin}{\code{CallbackSetUnfreeze$on_epoch_begin()}}
 \item \href{#method-CallbackSetUnfreeze-on_batch_begin}{\code{CallbackSetUnfreeze$on_batch_begin()}}
 \item \href{#method-CallbackSetUnfreeze-clone}{\code{CallbackSetUnfreeze$clone()}}
@@ -61,7 +60,7 @@ Creates a new instance of this \link[R6:R6Class]{R6} class.
 \subsection{Arguments}{
 \if{html}{\out{<div class="arguments">}}
 \describe{
-\item{\code{starting_weights}}{(\code{SelectorParam})\cr
+\item{\code{starting_weights}}{(\code{Select})\cr
 A \code{Select} denoting the weights that are trainable from the start.}
 
 \item{\code{unfreeze}}{(\code{data.table})\cr
@@ -79,16 +78,6 @@ Sets the starting weights
 \if{html}{\out{<div class="r">}}\preformatted{CallbackSetUnfreeze$on_begin()}\if{html}{\out{</div>}}
 }
 
-}
-\if{html}{\out{<hr>}}
-\if{html}{\out{<a id="method-CallbackSetUnfreeze-on_batch_end"></a>}}
-\if{latex}{\out{\hypertarget{method-CallbackSetUnfreeze-on_batch_end}{}}}
-\subsection{Method \code{on_batch_end()}}{
-Increment the batch counter (old)
-\subsection{Usage}{
-\if{html}{\out{<div class="r">}}\preformatted{CallbackSetUnfreeze$on_batch_end()}\if{html}{\out{</div>}}
-}
-
 }
 \if{html}{\out{<hr>}}
 \if{html}{\out{<a id="method-CallbackSetUnfreeze-on_epoch_begin"></a>}}