Reverting the change back - the original code was correct.