Probably fixed the failing test