Moved comment to right place