Fixes runtime in train/explode()

Thanks @Walter0o