Missed one line in test