AI-539 : clean up