Fixed compile error with internal class