Fixed return type on load function