Hotfixed my own code.