Fix class path

meh