Adding back a piece of code that somehow got deleted