fix test for page layout and remove dataset dependency