Corrected sample code to pass the correct type for timeout (time.Duration instead of int)