Fixed a silly error with class references