WIP: Restrict tests (Do not merge)