fixed code compilation issues in #5 module