Typing fix for register function

[AlemTuzlak]

Description

This PR fixes the issue with register function being spread over the remix context not working properly

Fixes #8

• [x] Bug fix (non-breaking change which fixes an issue)