#342 Fixed failing test