changed module definition
fixed removing the className