Fixed code that broke the build