fix error when running
IT WORKS FIRST TIME!