Fix tests for codebot, hopefully