Add forgotten test