Clean up after merge
Adding test to pass correlation test