Correction on test in Highway model