AI-499 : clean up