Small fix to set time out.