removed bug when restarting