Added basic check for using multistage