Fixed the build for me