minor changes to get it to compile