Add ugly tests