Fixed the bug which made the module fail.