#14 Update tests