Fix test that returns an ArgumentError