Fixes for agent stage