Fixes bug in loader refactor which prevented registration