Fixing compiler error