Fixes test_base.py so it becomes 100% covered