Fix build (hopefully)