[Fix] first() only returns the first element