Fixing their code...