Can we not use the select method in base.py instead of duplicating the code here?