Fixed small bug, thanks to demon_ix.