Making the check for both types of migrations
Signed-off-by: Siddharth Vinothkumar <siddharth.vinothkumar@citrix.com>