Fix f49be46c - added code to wrong part of file