Fix missing 'then'