diff --git a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub
index b1764f8bc..fef906630 100644
--- a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub
+++ b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub
@@ -8,7 +8,7 @@ from safeds.ml.nn.typing import ModelImageSize
  *
  * @param probability The probability that an input element gets zeroed out.
  */
- @Category(DataScienceCategory.ModelingQNeuralNetwork)
+@Category(DataScienceCategory.ModelingQNeuralNetwork)
 class DropoutLayer(
     probability: Float
 ) sub Layer {
diff --git a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub
index 3b995ad75..fe862c7e0 100644
--- a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub
+++ b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub
@@ -11,6 +11,7 @@ from safeds.ml.nn.typing import ModelImageSize
  * @param padding the padding of the pooling
  */
 @Experimental
+@Category(DataScienceCategory.ModelingQNeuralNetwork)
 class MaxPooling2DLayer(
     @PythonName("kernel_size") kernelSize: Int,
     stride: Int = -1,