Add method to set the default entity base