Fix missing callback parameter in keys module and register keys module to api