improve expected behaviour