Checking $stage parameter when methods are called