I guess this is ok, but it feels like it mirrors the actual code too much.