Update to support migrations

Signed-off-by: Ryan J. Peterson <rpeterson@serenitysoft.com>