Fixed error in test from rebasing