Fixed problem with code working only if run after original version has been run at some point.