adding condition for stage only tests