Python working except display debug. Tests are broken