Made this only work in Survival. Fixed a bug in the process.