Fix check_output call

Maybe someday I'll even learn Python!