Fixed compilation error due to shoter changes