Add timeout to integration test