Write more realistic tests