diff --git a/framework/libra-framework/sources/ol_sources/proof_of_fee.move b/framework/libra-framework/sources/ol_sources/proof_of_fee.move
index edbc41683..e1d8bb23d 100644
--- a/framework/libra-framework/sources/ol_sources/proof_of_fee.move
+++ b/framework/libra-framework/sources/ol_sources/proof_of_fee.move
@@ -90,7 +90,7 @@ module ol_framework::proof_of_fee {
   // from supply data.
   public fun genesis_migrate_reward(vm: &signer, nominal_reward: u64) acquires
   ConsensusReward {
-    if (signer::address_of(vm) != @ol_framework) return;
+    system_addresses::assert_ol(vm); // either 0x1 or 0x0
 
     let state = borrow_global_mut<ConsensusReward>(@ol_framework);
     state.nominal_reward = nominal_reward;