Return flat weight value if Germany shipping destination

Weight in DE

add_filter('pr_shipping_dhl_order_weight', 'custom_add_weight', 10, 2);
function custom_add_weight($total_weight, $order_id) {
	$order = wc_get_order( $order_id );
	if ($order) {
		$shipping_address = $order->get_address( 'shipping' );
		if( $shipping_address['country'] == 'DE' ) {
			return 30;
		}
	}

	return $total_weight;
}