Missed test fix