Compile fixes

some minor fixes to make the module compile again.