Fixed ConcurrentModificationException

You forgot some things ;)