Only use after_initialize to disable columns

Both tests pass without overriding columns().