fix for the failing test