Fixed missing references that weren't caught due to leftover pyc file allowing tests to keep working when they shouldn't have.