Forgot to copy this class over