fixing brokenness in test