IllegalArgumentException Fix