Fix build (oops, sorry)