Small timeout fix