Fixing test bug