Fix tests (fix #11)