#fix - custom class problem

Actually is not a problem at all lol

Signed-off-by: Ryne Cheow <rynecheow@gmail.com>