Add module type sanity check