Restore integration test