Fixes a nil class error