Add better test of behaviour