BZ-9549: class change

Signed-off-by: Paden <padenportillo@gmail.com>