Fixed failing test. Added another