Fixing class install process