fix failing test

This reverts commit 8ecd8f45b534f30f5ec2264e7f85b149b643235b.