Add missing line to test