Fixes broken tests
Fix array problem