Fixed class implementation error