Fix timeout in Integration Test