bug fixes so it compiles