Fixing typo

Moving block of code to the right function