Update tests (#6)