Fix code in wrong place