Integration test fixed