small changes to make it build