Update for Magma changes

Signed-off-by: Steven Downer <grinch@outlook.com>