Fix build error, hopefully