Fix missing line in test case