fixed code that got deleted