Set the timeout properly