Fixes for module 6