fixed a little bug and moved comment