Attempt to fix failing test