Trying to fix the failing test for limit