Fixed copy and past class name bug