Fix regression when using in with module attributes