Fixed class on bug report