Fix tests #1