Small Fix

Forgot to rename the classes...