Forgot to add the base_class on array too.