Added two lines of code that were completely missing!