Bug fix - minor timeout error