Fix build, hopefully