TST: Fixes allow mock entrypoints