Revert "Test fix"

This reverts commit 75ce08a587d31a651852517a8abbf338b3924a57.