Add assertion for array count