Fixed a bug with the warlock