Small change in java code