Fix for sub class issue