addressing ashutohs comments
fixing compile issue