Fixed issue with tests & __client__ thing.

This is what's causing that `__client__` error that you're getting.