Slightly cleaner mock setup