Fix class rename from 2 commits ago