diff --git a/src/local_layer.h b/src/local_layer.h
index 8ef0dbad393..45e02a5c0a1 100644
--- a/src/local_layer.h
+++ b/src/local_layer.h
@@ -15,7 +15,7 @@ extern "C" {
 #ifdef GPU
 void forward_local_layer_gpu(local_layer layer, network_state state);
 void backward_local_layer_gpu(local_layer layer, network_state state);
-void update_local_layer_gpu(local_layer layer, int batch, float learning_rate, float momentum, float decay);
+void update_local_layer_gpu(local_layer layer, int batch, float learning_rate, float momentum, float decay, float loss_scale);
 
 void push_local_layer(local_layer layer);
 void pull_local_layer(local_layer layer);