Fix ArgumentError