Fix the unit-tests for the change made in page_total