WA but I think this is a correct solution