Fix custom register