Fixing outdated test assertion