Ändrade en annan del i koden som var fel

Ändrade en annan del i koden som gjorde att det blev fel..Nu funkar den
bra