Move check to canRun method in ab test module