Fixing Test Error