Fixes regression