Fixed 'Agent button' not showing up on WorkspaceView
Added more tests to ToolBox to ensure its behaviour as it was looking like it was weird