Minor bug fix (on level restart)