Moved comment to correct class