Fix timeout error

Fix timeout error