a small modification

Just here, thanks to DarkSun.