Fix test module inherits