Fix IllegalArgumentException