Update sanity check on Main.cpp