Fix the codebase for ToolLabs