Fixes broken build