Fixed a crash on load

Fixed a crash on load due to forgotten code