Little code change