Fixes a regression