Adding module check on import