Made the test example work with nicks changes