Fix registration module
Signed-off-by: zurcxer <zurcxer@gmail.com>