fixed code that i forgot to remove