Separate this into 2 unit tests, one for each register