Fixing register tests