Fixed the where call to return an array