Minor editing change in code