Updated tests to include timeout call