Fixing some tests after big refact of idealist component