[minor] CS Fixes
Added missing use statement