Fixed issue including another class