cleanup per pr comments
fix comment