AI-621 - fixes