Partially fixed bug when restarting gameplay.