Mock only certain calls.

#7