useLayout issues with useState