using the correct class