fix code that probs fixed on my desktop