Fixes breakage

:(