Fixes the refactor fix.